[Swift｜코드카타] (프로그래머스) 입문 #25. 나누어 떨어지는 숫자 배열

🔴 문제

https://school.programmers.co.kr/learn/courses/30/lessons/12910

 

 

🔵 풀이 1 - for...in, append, sorted

for...in 구문에서 element를 divisor로 나눠보고, 나누어 떨어지는 element가 있다면 새로운 배열에 append하는 방식으로 풀면 될 것 같다.

func solution(_ arr:[Int], _ divisor:Int) -> [Int] {
    var newArr: [Int] = []
    
    for i in arr {
        if i % divisor == 0 {
            newArr.append(i)
        }
    }
    
    if newArr == [] { return [-1] }
    
    return newArr.sorted() // 오름차순으로 정렬하여 반환
}

 

 

🔵 풀이 2 - filter, sorted

특정 조건을 만족하는 element를 필터링하여 배열을 반환하는 것이므로, 고차함수 filter를 사용할 수 있다.

func solution(_ arr:[Int], _ divisor:Int) -> [Int] {
    let result = arr.filter{ $0 % divisor == 0}.sorted()
    return result.isEmpty ? [-1] : result
}

 

<관련 문서>

- filter 개념 정리 : [Swift] 고차함수 - map, filter, reduce

- filter 사용 예제 : [Swift｜코드카타] (프로그래머스) 입문 #14. 약수의 합 (filter, reduce, squareRoot)