[프로그래머스] (Lv.1) K번째수 *

문제

---

배열 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 함수를 작성해주세요.

 

 

-  출력 예시  -

정답 코드

---

<  내 정답 코드  >

class Solution {
    public int[] solution(int[] array, int[][] commands) {
        List<Integer> list = new ArrayList<>();
        int idx = 0;
        
        for (int[] command : commands) {
            List<Integer> tmp = new ArrayList<>();
            for (int i = command[0] - 1; i < command[1]; i++) {
                tmp.add(array[i]);
            }
            Collections.sort(tmp);
            list.add(tmp.get(command[2] - 1));
        }
        
        int[] answer = new int[list.size()];
        for (Integer i : list) {
            answer[idx++] = i;
        }
        
        return answer;
    }
}

 

<  타인 답변 코드  >

class Solution {
    public int[] solution(int[] array, int[][] commands) {
        int[] answer = new int[commands.length];

        for(int i = 0; i < commands.length; i++){
            int[] temp = Arrays.copyOfRange(array, commands[i][0]-1, commands[i][1]);
            Arrays.sort(temp);
            answer[i] = temp[commands[i][2]-1];
        }

        return answer;
    }
}

이것을 주의하자!

---

-  반환해야 하는 배열이 어떤 크기가 될지 모른다 (하지만 잘못된 생각). 그러면 일단 리스트로 만든 다음 나중에 배열로 전환해야 한다. 그래서 맨 처음 ArrayList 를 만들었다. 그리고 for 문을 통해 배열을 하나씩 꺼내와 해당 조건에 맞는 원소들을 하나씩 tmp 리스트에 저장한다. 작업이 끝나면 정렬한 다음 원하는 조건의 원소를 뽑아 result 리스트에 저장한다. 그렇게 모든 작업 후에 배열로 바꿔서 반환한다.

 

-  타인 답변 코드를 보면 "반환해야 하는 배열의 크기를 알 수 있다". 새롭게 반환해야 하는 배열의 원소 개수는 int [][] commands 의 개수와 동일하기 때문이다.

 

-  Arrays.copyOfRange( ) 메서드를 통해 원하는 범위만큼 배열을 복사할 수 있다. 그리고 복사한 배열에서 원하는 값을 뺀 뒤 정답 배열에 넣는다. 이 방법이 훨씬 깔끔하다.