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

사진을 클릭하면 해당 문제로 이동

💡풀이

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

// commands 별로 분기한다. (i j k)
// i ~ j 번째 배열로 잘라낸다.
// 잘라낸 배열을 오름차순 정렬한다.
// k 번째 원소를 찾아 answer에 저장한다.

class Solution {
    public int[] solution(int[] array, int[][] commands) {
        List<Integer> answer = new ArrayList<>();

        for (int[] command : commands) {
            int i = command[0] - 1; // index로 변환하기 위해 각각 -1 적용
            int j = command[1] - 1;
            int k = command[2] - 1;
            
            int[] arr = Arrays.copyOfRange(array, i, j+1);
            Arrays.sort(arr);

            answer.add(arr[k]);
        }

        return answer.stream().mapToInt(Integer::intValue).toArray();
    }
}

---

📖새로 배운 부분

배열 복사 관련 메서드를 글로 정리해 보았습니다.

[Java] 배열 복사 메서드 간단 정리

---

GitHub - okjunghyeon/Programmers_CodingTest: 프로그래머스 관련 코딩테스트 문제를 풀이한 저장소입니다.