[프로그래머스] K번째수 (lv.1)

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

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

 

[풀이 과정]

 

1. 여러 개의 명령어(i, j, k)가 들어 올 수 있으므로, 기존의 배열을 수정하면 안된다.

2. 명령어의 길이는 항상 3이므로, 예외 처리를 할 필요가 없다.

3. i부터 j까지 배열을 복사하고 정렬한다.

4. 정렬한 배열에서 k번째 수를 찾는다.

 

[소스 코드]

 

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class KthNumber {
    public static void main(String[] args) {
        int[] array = {1, 5, 2, 6, 3, 7, 4};
        int[][] commands = {
                {2, 5, 3},
                {4, 4, 1},
                {1, 7, 3}
        };

        System.out.println(Arrays.toString(solution(array, commands)));
    }

    private static int[] solution(int[] array, int[][] commands) {
        List<Integer> temps = new ArrayList<>();

        for (int[] command : commands) {
            int[] temp = Arrays.copyOfRange(array, command[0] - 1, command[1]);
            Arrays.sort(temp);
            temps.add(temp[command[2] - 1]);
        }

        int[] answer = new int[temps.size()];

        for (int i = 0; i < answer.length; i++) {
            answer[i] = temps.get(i);
        }

        return answer;
    }
}