[Lv.1] K번째 수

문제 설명

배열 `array`의 `i`번째 숫자부터 `j`번째 숫자까지 자르고 정렬했을 때, `k`번째에 있는 수를 구하려 합니다.

예를 들어 `array`가 `[1, 5, 2, 6, 3, 7, 4]`, `i = 2`, `j = 5`, `k = 3`이라면 `array`의 2번째부터 5번째까지 자르면 `[5, 2, 6, 3]`입니다. 1에서 나온 배열을 정렬하면 `[2, 3, 5, 6]`입니다. 2에서 나온 배열의 3번째 숫자는 5입니다. 배열 `array`, `[i, j, k]`를 원소로 가진 2차원 배열 `commands`가 매개변수로 주어질 때, `commands`의 모든 원소에 대해 앞서 설명한 연산을 적용했을 때 나온 결과를 배열에 담아 return 하도록 solution 함수를 작성해주세요.

 

 

📌부분 배열 구하기

java.util 패키지의 `Arrays` 클래스에는 배열을 다루기 위한 메소드가 포함되어 있다. 모든 메소드는 static 메소드로, 객체를 생성하지 않고도 사용할 수 있다. java.util 패키지에 포함되므로 반드시 import문으로 패키지를 불러온 후 사용한다.

 

특정 길이만큼 새로운 배열로 복사하여 반환

`copyOf()`

첫번째 인자로 원본 배열을 전달받고 두번째 인자로 복사할 요소의 개수를 받는다. 

 

import java.util.Arrays;

// ...

int[] arr = {1, 2, 3, 4, 5};
int[] arr2 = Arrays.copyOf(arr1, 3);

// arr2 = {1, 2, 3};

 

 

 

특정 범위에 해당하는 요소만 새로운 배열로 복사하여 반환

`copyOfRange()`

첫번째 인자로 원본 배열을 전달받고 두번째 인자로 시작 인덱스를, 세번째 인자로 마지막 요소가 될 인덱스를 전달받는다.

 

int[] arr1 = {1, 2, 3, 4, 5};
int[] arr2 = Arrays.copyOfRange(arr1, 2, 4);

// arr2 = {3, 4}

 

 

 

 

📌배열 정렬하기

`Arrays` 클래스의 `sort()` 메소드는 배열을 오름차순으로 정렬해준다.

 

int[] arr = {2, 5, 3, 4, 1};
Arrays.sort(arr);

// arr = {1, 2, 3, 4, 5};

 

 

 

📌풀이

1. 주어진 배열 `array`를 i부터 j까지 자른 뒤 k번째의 원소를 구한다. 해당 원소를 answer 배열에 담아 반환한다.

2. `commands`의 인덱스 0은 i, 1은 j, 2는 k가 된다. 

3. 임시 배열 temp에 array를 i부터 j까지 자른 부분 배열을 넣은 뒤, temp의 k번째 원소를 answer에 넣는다.

 

단, i 또는 k번째 원소를 구하는 것은 인덱스 k - 1 의 원소를 구하는 것과 동일하므로 반드시 1을 빼준다.

 

 

 

전체 코드

 

import java.util.*;

class Solution {
    public int[] solution(int[] array, int[][] commands) {
        int[] answer = new int[commands.length];
        int[] temp = {};
        
        // i부터 j까지 자른 뒤 k번째의 원소를 구한다.
        // commands의 인덱스 0은 i, 1은 j, 2는 k
        // 임시 배열 temp에 array를 i부터 j까지 자른 부분 배열을 넣은 뒤, temp의 k번째 원소를 answer에 넣는다.
        
        for(int i = 0; i < commands.length; i++){
            temp = Arrays.copyOfRange(array, commands[i][0] - 1, commands[i][1]);
            Arrays.sort(temp);
            answer[i] = temp[commands[i][2] - 1];
        }
        
        
        return answer;
    }
}

 

 

 

 

 

 

 

프로그래머스