[Algo Rhythm🕺💃] 프로그래머스 고득점 Kit K번째 수

문제 설명

문제는 아래 링크에서 확인하시길 바랍니다.

https://programmers.co.kr/learn/courses/30/lessons/42748

코딩테스트 연습 - K번째수

나는 어떻게 풀었나

import java.util.Arrays;

class Solution {
    public int[] solution(int[] array, int[][] commands) {
        int[] answer = new int[commands.length];
        
        for(int a = 0; a < commands.length; a++) {
            int i = commands[a][0];
            int j = commands[a][1];
            int k = commands[a][2];
            
            int[] temp = Arrays.copyOfRange(array, i - 1, j); // 1. slice
            Arrays.sort(temp); // 2. sort
            answer[a] = temp[k - 1];
        }
        
        return answer;
    }
}

 

나는 왜 저렇게 풀었나

지문을 읽으면 지문에서 이미 아래와 같은 알고리즘을 명시하고 있다는 것을 알 수 있다.

 

1. 배열의 i번째부터 j번째 원소까지 자른다.
2. 자른 배열을 sort한다.
3. Sort한 배열의 k번째 원소를 구한다.

그래서 먼저 배열을 자르는 알고리즘을 생각했다. 직접 코드를 구현할 수도 있지만 'slice 정도의 기능은 java에서 이미 제공해주지 않을까?' 라는 생각에 Arrays의 javadoc을 읽어봤다.

https://docs.oracle.com/en/java/javase/11/docs/api/java.base/java/util/Arrays.html

 

아쉽게도 slice라는 이름의 method는 존재하지 않았다. 하지만 같은 기능을 하는 copyOfRange라는 method를 발견했다. 그래서 이를 사용해 slice한 뒤에 sort를 하고 k번째 (1-based 일때) 원소를 answer에 담아줘서 해결했다.

 

나는 무엇을 배웠나

늘 생각하는 것이지만 Java의 강점은 역시 Javadoc이다. Javadoc을 읽는 것이 느리고 답답할 수 있지만 실력을 올리기에는 최고라고 생각한다. 뿐만 아니라 이번 기회에 slice기능을 수행하는 copyOfRange를 배웠는데 잊지 않고 잘 활용해야겠다.