[Algorithm/JAVA][프로그래머스]K번째 수

[프로그래머스] K번째 수

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

문제 설명

배열 array의 i번째 숫자부터 j번째 숫자까지 자르고 정렬했을 때, k번째에 있는 수를 구하려 합니다.
예를 들어 array가 [1, 5, 2, 6, 3, 7, 4], i = 2, j = 5, k = 3이라면

1. array의 2번째부터 5번째까지 자르면 [5, 2, 6, 3]입니다.
2. 1에서 나온 배열을 정렬하면 [2, 3, 5, 6]입니다.
3. 2에서 나온 배열의 3번째 숫자는 5입니다.

배열 array, [i, j, k]를 원소로 가진 2차원 배열 commands가 매개변수로 주어질 때, commands의 모든 원소에 대해 앞서 설명한 연산을 적용했을 때 나온 결과를 배열에 담아 return 하도록 solution 함수를 작성해주세요.

제한사항

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

Code

import java.util.Arrays;
class Solution {
    public int[] solution(int[] array, int[][] commands) {
        int[] answer = new int[commands.length];
        for(int i=0;i<commands.length;i++){
            int a,b,c;
            a=commands[i][0];
            b=commands[i][1];
            c=commands[i][2];
            int[] temp = Arrays.copyOfRange(array,a-1,b);
            Arrays.sort(temp);
            answer[i]=temp[c-1];
        }
        return answer;
    }
}

Code 설명

answer의 크기는 commands의 크기와 동일하고 commands의 크기만큼 반복문을 돌아야 한다.
문제설명에서 i는 a, j는 b, k는 c로 받아서 input인 array에 a번째부터 b번째까지 복사하여 temp로 옮기고 temp를 정렬해서 c번째를 answer[i]에 넣어주었다.

배운점

Java를 2년전에 잠깐 학과에서 배우고 사용을 안하다가 이번에 알고리즘 공부도 할 겸 자바공부를 시작해서 아직 자바 사용이 낯설다.
기존에 쓰던 C/C++과 달리 Java는 그냥 배열에서도 사용할 수 있는 편한 메소드들이 많다는 것을 느끼게 되었다.

1. 배열선언
   자바는 배열을 선언할 때 int[] answer = new int[commands.length] 와 같이 할당을 해주어야 한다.
2. java.util.Arrays
   자바의 util 패키지는 정말 자주 쓰이고 유용한 것들이 많다고 한다. 보통 import java.util.*를 많이 사용한다. 하지만 아직 뭐가 있는지 모르므로 필요한게 나올때마다 하나씩 import해줘서 사용해 보려고 한다.
   Arrays클래스는 이름에서 알 수 있듯이 배열을 다루는 메소드들이 포함되어 있다.
   이번에 사용한 메소드는 copyOfRange(), sort() 이다.

copyOfRange(arr,startidx,endidx)
이 메소드는 첫 번째 매개변수로 복사할 원본 배열을 받고, 두 번째 매개변수로 시작 인덱스, 세 번째 매개변수로 마지막 복사될 배열 인덱스+1을 받아서 원본 배열과 같은 타입의 복사된 새로운 배열을 반환해준다.

sort(arr)
이 메소드는 매개변수로 배열을 받고 배열을 오름차순으로 정렬해준다.