연습#2 K번째수

안녕하세요.

도저히 기본기가 부족하다는 생각에 핀의 추천으로 프로그래머스 알고리즘Kit을 풀기 시작했습니다.

 

정렬파트의 K번째수 문제를 풀었습니다.

 

import java.util.*;

class Solution {
    public int[] solution(int[] array, int[][] commands) {     
        int[] res = new int[commands.length];
        for (int i=0; i<commands.length; i++) {     
            int[] command = commands[i];
            int start = command[0];
            int end = command[1];
            int pick = command[2];

            List<Integer> picks = new ArrayList();
            for(int j=start-1; j<=end-1; j++) {
                picks.add(array[j]);
            }

            picks.sort(Comparator.naturalOrder());
            
            res[i] = picks.get(pick-1);
        }
        
        return res;
    }
}

 

 

O(n^2)의 시간복잡도를 가진 풀이입니다.

 

이 문제를 풀면서 배운 점이 있습니다.

1. import는 *로 한다.
2. 답의 수는 commands의 행의 개수이므로, 배열의 크기는 정적임을 알 수 있다. 

 

뭔가 하나씩 더 배우네요. 파이팅입니다!