빙응의 공부 블로그

[Programmers]Lv.1 K번째 수 본문

Argorithm

[Programmers]Lv.1 K번째 수

빙응이 2024. 1. 2. 20:21


📝풀이

배열 정렬 문제 가볍게 sort를 썼고 사실상 구현 문제이다. 

나는 배열 저장에 ArrayList를 썼는데 안써도 상관 없을 것 같다.

import java.util.ArrayList;
import java.util.Collections;

class Solution {
    public int[] solution(int[] array, int[][] commands) {
        ArrayList<Integer> al = new ArrayList<>();
        int[] answer = new int[commands.length];
        
        for (int i = 0; i < commands.length; i++) {
            al.clear(); 

            for (int j = commands[i][0] - 1; j < commands[i][1]; j++) {
                al.add(array[j]);
            }

            Collections.sort(al); 
            answer[i] = al.get(commands[i][2] - 1);
        }

        return answer;
    }
}