[JAVA] K번째 수

문제

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

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

내 풀이

import java.util.*;

class Solution {
    public int[] solution(int[] array, int[][] commands) {
        int[] answer = new int[commands.length];
        int index = 0;
        List<Integer> list = new ArrayList<>();
        Arrays.stream(array).forEach(i -> list.add(i));
        
        for (int[] command: commands) {
            List<Integer> subList = new ArrayList<>(list.subList(command[0]-1, command[1]));
            Collections.sort(subList);
            answer[index++] = subList.get(command[2]-1);
        }

        
        return answer;
    }
}

subList 의 주의점은 원본 리스트에 종속적이므로 새로운 리스트 객체를 생성해 할당해주는 것이 필요했다.

위처럼 ArrayList 객체를 생성해 할당하지 않으면 sort시에 원본 list에도 소팅이 적용되어 2번째 loop 부터 값이 

원하는대로 나오지 않는다.

 

알게 된 것

subList의 주의해야할 점을 알았다.

 

피드백

int[] temp = Arrays.copyOfRange(array, commands[i][0]-1, commands[i][1]);
Arrays.sort(temp);

List 객체말고 Arrays class 를 활용해 array에 대한 값의 복사 배열을 생성할 수 있다.

위처럼 하면 List 객체들의 생성없이 단순 배열로만 해결이 가능할 것이다.

https://docs.oracle.com/javase/7/docs/api/java/util/Arrays.html#copyOfRange(int[],%20int,%20int) 

Arrays (Java Platform SE 7 )