K번째수(java)

문제설명

생각하기

1. commands의 길이만큼 반복하며 i,j까지 돌며 원소를 뽑기
2. 뽑은 원소를 정렬시키기
3. 그 중 k번째 수 찾기

내 풀이

import java.util.*;
class Solution {
    public int[] solution(int[] array, int[][] commands) {
        int[] ans = new int[commands.length];
        ArrayList<Integer> list = new ArrayList<>();
        
        for( int i=0; i<commands.length; i++){
            for(int j=commands[i][0]-1; j<commands[i][1]; j++){
                list.add(array[j]);
            }
            Collections.sort(list);
            ans[i] = list.get(commands[i][2]-1);
            list.clear();
        }
        
        return ans;
    }
}

ArrayList를 사용하여 list에 i번째부터 k까지의 수를 담고 정렬시킨다.
그 중 k번째 수를 ans[]에 담는다.
리스트를 초기화시키고 반복