나누어 떨어지는 숫자 배열

25. 나누어 떨어지는 숫자 배열

class Solution {
    fun solution(arr: IntArray, divisor: Int): IntArray {
        var answer = intArrayOf()

        arr.forEach {
            if (it % divisor == 0) {
                answer += it
            }
        }
            answer.sort()

        if (answer.isEmpty()) {
            answer += -1
        }

        return answer
    }
}

 

내가 풀던 식은 뭘 해도 첫번째 테스트에서 5밖에 안나와서 결국 다른분 코드를 참고했다.

다른건 거의 같은데 이분은 forEach 를 쓰신 점이 달랐다.

그런데 이걸 봐도 내 이전 코드가 왜 틀렸는지는 아직 잘 모르겠다...

 

forEach는 배열 또는 리스트에서 사용 가능한 반복문이다.

arr를 반복해서 divisor로 나눈 나머지가 0이면 answer에 더해준다. 그리고 sort로 정렬해준다.

그리고 나머지가 0인 숫자가 하나도 없으면 -1 를 더해준 뒤 return한다.