프로그래머스[Java] - K번째수(정렬)

https://programmers.co.kr/learn/courses/30/lessons/42748

코딩테스트 연습 - 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++) {
			// 새로운 배열의 크기 -> commands[i][1] ~ commands[i][0]까지 자른 후 정렬하므로 두 값 빼고 +1
			int[] arr = new int[commands[i][1] - commands[i][0] + 1];
			int arrIndex = 0;
			
			// commadns의 [i][0]-1 ~ [i][1]까지 잘라서 array 배열의 요소 값을 arr에 저장하기.
			// 예제대로 2번째 ~ 5번째라면 배열의 인덱스상 1, 2, 3, 4 값이 들어가야 하므로 -1 해줌.
			for(int j=commands[i][0]-1; j<commands[i][1]; j++) {
				arr[arrIndex++] = array[j];
			}
			
			Arrays.sort(arr);	// 배열 정렬
			answer[i] = arr[commands[i][2] - 1];
		}
		return answer;
	}
}

풀이

 

문제에서 주어진 대로 그대로 구현했다.

commands[][] 배열의 각 행마다 새로운 배열(arr)을 만든다.

새로운 배열의 크기는 주어진 commands의 첫번째 열 - 0번째열 + 1 을 하면 되는데,

2번째 ~ 5번째는 5 - 2 + 1 = 4

4번째 ~ 4번째는 4 - 4 + 1 = 1

1번째 ~ 7번째는 7 - 1 + 1 = 7

위와 같이 되므로, 새로운 배열의 크기를 정해준다.

 

commands[i][0] ~ commands[i][1] 번째 수를 새로운 배열(arr)에 넣고, 

새로운 배열(arr)을 정렬한 후 arr[commands[i][2]-1] 을 answer 배열에 넣어주면 된다.

배열의 인덱스는 0부터이므로, 한칸씩 당겨서 -1을 해줘야 한다.

[1, 5, 2, 6, 3, 7, 4] 를 2번째 ~ 5번째 자른 후 정렬 => [5, 2, 6, 3] => [2, 3, 5, 6]

2번째 ~ 5번째 이지만, 배열의 인덱스로는 1, 2, 3, 4 가 된다.