프로그래머스 - K번째 수 JAVA

C++로 풀다보니 C++구현에만 익숙해서 자바로도 풀어보려고 한다. 정렬 문제를 풀어보자

 

 

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

코딩테스트 연습 - K번째수

 

 

1.

아무것도 모르고 푼거

우우,,

import java.util.*;
 
 
class Solution {
    public int[] solution(int[] array, int[][] commands) {
        int[] answer = new int[commands.length];
 
        for(int s=0;s<commands.length; s++){
            int i = commands[s][0];
            int j = commands[s][1];
            int k = commands[s][2];
 
            int[] tmp = new int[j-i+1];
            int idx = 0;
            while(idx<tmp.length){
                tmp[idx]=array[i-1];
                idx++;
                i++;
            }
 
            Arrays.sort(tmp);
            System.out.println();
            answer[s]=tmp[k-1];
 
        }
 
 
        return answer;
    }
}
 

 

2. 다른사람 풀이

Arrays.copyOfRange 이용해서 푼거!

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[] tmp = Arrays.copyOfRange(array,commands[i][0]-1,commands[i][1]);
            Arrays.sort(tmp);
            answer[i]=tmp[commands[i][2]-1];
        }
        
        return answer;
    }
}

깔끔하게 풀린다.