프로그래머스: K번째 수

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

 

2019년 초 풀이

내가 푼 방법이 효율적이라고 할 수는 없겠지만, 이렇게 푸는 사람도 있다 정도로 참고만 하면 좋겠다.

나는 연산(배열을 자른 후 정렬, n번째 수 찾기)을 위해서 ArrayList를 사용했다. 이 때 ArrayList 대신 Vector 혹은 일반 배열을 사용해도 상관없다.

import java.util.ArrayList;
import java.util.Collections;
 
class Solution {
    public int[] solution(int[] array, int[][] commands) {
        int[] answer = new int[commands.length];
 
        ArrayList<Integer> temp = new ArrayList<>();
 
        for (int i = 0; i < commands.length; i++) {
            for (int j = 0; j < commands[i].length; j++) {
                if (j == 0) {
                    for (int k = commands[i][j] -1; k < commands[i][1]; k++) { 
                        temp.add(array[k]);
                    }
                    Collections.sort(temp);
                }
            }
            answer[i] = temp.get(commands[i][2] - 1);
            temp.clear();
        }
 
        return answer;
    }
}

 

어서와! 자료구조와 알고리즘은 처음이지? 수강 후 풀이

def solution(array, commands):
    answer = []
    for i in commands:
        start = i[0] - 1
        end = i[1]
        k = i[2] - 1
        
        temp = array[start:end]
        temp.sort()
        answer.append(temp[k])
        
    return answer

 

이렇게 놓고 비교하니까 확실히 다르긴 하네.