[Algorithm] 프로그래머스 K번째 수

문제

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

 

풀이

1. 커맨드 별로 진행한다.

2. 배열의 i번째부터 j번째까지 리스트에 담고 정렬한다. k번째 수를 정답 배열에 추가한다.

3. 로직이 시작될 때, 이전 연산이 남지 않도록 리스트를 비운다.

 

전체 소스코드

import java.util.*;

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

 

다른 풀이

리스트를 사용하지 않고 내부에 배열 생성해서 배열 카피 후 정렬, 그리고 k번째 수 찾음.

import java.util.*;

class Solution {
    public int[] solution(int[] array, int[][] commands) {
        int[] answer = new int[commands.length];
        
        for (int i = 0; i < commands.length; i++) {
            int[] command = commands[i];
            int[] parts = Arrays.copyOfRange(array, command[0]-1, command[1]);
            Arrays.sort(parts);
            answer[i] = parts[command[2]-1];
        }
        
        return answer;
    }
}