Android/Algorithm

[프로그래머스 / Kotlin] 나누어 떨어지는 숫자 배열

정생강 2024. 3. 6. 10:00

문제 설명

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

- 제한 사항

arr은 자연수를 담은 배열입니다.

정수 i, j에 대해 i ≠ j 이면 arr[i] ≠ arr[j] 입니다.

divisor는 자연수입니다.

array는 길이 1 이상인 배열입니다.

 

풀이 과정

해당 문제를 구현하는데에는 크게 어려움을 겪지 않았다. 3가지 함수만 알고 있다면 이 문제를 쉽게 해결이 가능한데 배열을 오름차순으로 정렬하는 sort(),  배열의 크기를 구하는 size, 배열이 비어있는지 확인하는 isEmpty() 이다. 해당 문제는 입력 받은 배열이나 answer배열 둘 중에 하나를 오름차순 처리하는 과정이 필요하는데 나는 arr배열을 오름차순 하는걸로 택하였다. 그 후 반복문을 통해 divisor로 나누어 떨어지는 수를 따로 모아 리턴하였고  answer 배열이 비어있으면 -1을 넣어서 리턴하는 조건문 또한 설정해 두었다.

 

내가 푼 코드

class Solution {
    fun solution(arr: IntArray, divisor: Int): IntArray {
        var answer = intArrayOf()
        arr.sort()
        
        for(i in 0 until arr.size){
            if(arr[i] % divisor == 0) answer += arr[i]
        }
        if(answer.isEmpty()) answer += -1
        
        return answer
    }
}

 

후기

오늘 사용한 함수들은 자주 사용되기 때문에 꼭 기억해 둘 것.