[프로그래머스] 나누어 떨어지는 숫자 배열 .with Swift

--------------------문제 설명--------------------

array의 각 element 중 divisor로 나누어 떨어지는 값을 오름차순으로 정렬한 배열을 반환하는 함수, solution을 작성해주세요.
divisor로 나누어 떨어지는 element가 하나도 없다면 배열에 -1을 담아 반환하세요.

 

--------------------제한 사항--------------------

• arr은 자연수를 담은 배열입니다.
• 정수 i, j에 대해 i ≠ j 이면 arr[i] ≠ arr[j] 입니다.
• divisor는 자연수입니다.
• array는 길이 1 이상인 배열입니다.

-------------------- 풀이 --------------------

func solution(_ arr:[Int], _ divisor:Int) -> [Int] {
    var result: [Int] = []
    for i in arr{
        if i % divisor == 0 {
            result.append(i)
        }
    }
    result.sort(by: <)
    if result.isEmpty { result.append(-1) }
    return result
}

 

제가 만든 로직 입니다.

3번 줄 부터 for문을 열어 arr만큼 반복을 합니다.

4번 줄에서 i에서 divisor를 나눈 값이 0인 경우는  정확하게 나누어 떨어지는 값만 구분하여

5번 줄에서 result배열에 해당값을 추가 합니다.

그리고 8번줄에서 결과로 나온 배열을 오름차순으로 정렬해 줍니다.

다만 9번줄과 같이 result배열에서 값이 비워져 있는경우는 -1을 배열에 넣고

10번 라인에서 결과값을 반환해주면 해당 문제는 종료됩니다.