[프로그래머스 Lv1] k번째수 (Java)

문제

프로그래머스

---

코드

 

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

public class Solution29_2 {
	public static int[] solution(int[] array, int[][] commands) {
		int[] answer = {};

		ArrayList<Integer> list;
		ArrayList<Integer> result = new ArrayList<>();

		for (int i = 0; i < commands.length; i++) {
			list = new ArrayList<>();
			for (int j = commands[i][0] - 1; j < commands[i][1]; j++) {
				list.add(array[j]);
			}
			Collections.sort(list);

			result.add(list.get(commands[i][2] - 1));
		}

		answer = new int[result.size()];

		for (int i = 0; i < result.size(); i++) {
			answer[i] = result.get(i);
		}

		return answer;
	}

	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 } };
		System.out.println(Arrays.toString(solution(array, commands)));
	}
}

---

경험

 

인덱스는 0부터 시작이라서 문제에서 말하는 2~5까지 슬라이싱하는 게

1~4까지 슬라이싱을 하라는 뜻인 것에서 약간 버벅였다.