[programmers/Swift] 나누어 떨어지는 숫자 배열

programmers.co.kr - 코딩테스트연습 - Lv.1 - 연습문제 - 나누어 떨어지는 숫자 배열

코딩테스트 연습 - 나누어 떨어지는 숫자 배열

Constraints :

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

 

Solution.swift :

//
//  Created by Yongwoo Marco on 2022/05/02.
//  Copyright © 2022 Yongwoo Marco Kim. All rights reserved.
//

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

print(solution([5, 9, 7, 10], 5)) // [5, 10]
print(solution([2, 36, 1, 3], 1)) // [1, 2, 3, 36]
print(solution([3,2,6], 10)) // [-1]

How I tried this :
딱 하나 빼고 통과해서 고민해보니 `divisor`는 자연수라고 알려주지만 

arr에 0이 들어있을 수 있는 점은 따로 얘기하지 않았다.

0이 아닌 경우까지 포함하니 문제 없이 통과했다.

 

문제에 관한 모든 저작권 : https://programmers.co.kr/