안드 공부를 해볼까?

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

알고리즘/프로그래머스

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

문바리 2022. 5. 5. 10:50
728x90

1. 문제 파악

i부터 j까지 배열을 자르고 정렬한 후 k번째 숫자를 구하는 문제다.

문제에서 알려준대로 한다면 어려운 부분은 없다.

2. 구현

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++){
            //시작하는 곳
            int start = commands[i][0];
            //종료되는 곳
            int end = commands[i][1];
            //정렬 후 찾아야할 인덱스
            int idx = commands[i][2];
            //정렬한 값을 저장하는 배열
            int[] sortArr = new int[end-start+1];
            
            //배열에 값 추가
            for(int j = 0; j<sortArr.length; j++){
                sortArr[j] = array[start-1];
                start++;
            }
            
            //정렬 후 값 찾기
            Arrays.sort(sortArr);
            answer[i] = sortArr[idx-1];
        }
        return answer;
    }
}

따로 설명할 필요가 없다. 정말 문제 그대로 구현하면 끝이다.

3. 마무리

코테 문제가 이런식으로만 나온다면 얼마나 행복할까...😂

반응형
Comments