[프로그래머스 Lv.1] K번째 수 (정렬)

[프로그래머스 Lv.1] K번째 수 (정렬)

파이썬 풀이

def solution(array, commands):
    answer = []
    for c in commands:
        nums = array[c[0] - 1:c[1]]
        nums.sort()
        answer.append(nums[c[2] - 1])
    return answer

자바 풀이

import java.util.*;

class Solution {
    public int[] solution(int[] array, int[][] commands) {
        int[] answer = new int[commands.length];
        
        int index = 0;
        for(int[] c : commands){
            int[] nums = Arrays.copyOfRange(array, c[0] - 1, c[1]);
            Arrays.sort(nums);
            answer[index] = nums[c[2] - 1];
            index += 1;
        }
        
        return answer;
    }
}

• Arrays.copyOfRange()을 이용해서 배열을 slice함
• Arrays.sort()로 slice한 배열을 오름차순 정렬함