프로그래머스 알고리즘, K번째 수

목표 (문제)

2차배열인 commands 의 값마다 [i][0]과 [i][1]번을 시작과 끝으로 array 배열에서 분리 후,
오름차순 정렬 하여 command[i][2]의 index에 해당하는 값들을 return 한다.

조건

• array의 길이는 1 이상 100 이하입니다.
• array의 각 원소는 1 이상 100 이하입니다.
• commands의 길이는 1 이상 50 이하입니다.
• commands의 각 원소는 길이가 3입니다.

코드

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class Main {

    public static void main(String[] args) {
        int[] array = { 1, 5, 2, 6, 3, 7, 4 };
        int[][] commands = { { 2, 5, 3 }, { 4, 4, 1 }, { 1, 7, 3 } };

        for (int val : solution(a, b)) {
            System.out.print(val + ",");
        }
    }

    public static int[] solution(int[] array, int[][] commands) {
        List<Integer> answer = new ArrayList<>();

        for (int cmdIndex = 0; cmdIndex < commands.length; cmdIndex++) {
            int from = --commands[cmdIndex][0];
            int to = commands[cmdIndex][1];

            int[] splitArray = Arrays.copyOfRange(array, from, to);
            Arrays.sort(splitArray);

            int selectedIndex = commands[cmdIndex][2] - 1;
            answer.add(splitArray[selectedIndex]);
        }

        return answer.stream().mapToInt(i -> i).toArray();
    }
}