[프로그래머스] K번째수 Python, C++

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

프로그래머스

 

(1) C++

#include <string>
#include <vector>
#include <string.h>
#include <algorithm>

using namespace std;

vector<int> solution(vector<int> array, vector<vector<int>> commands) {
    vector<int> answer;
    for(int i = 0; i<commands.size(); i++){
        vector<int> array2;
        for(int j = commands[i][0] - 1; j < commands[i][1]; j++){
            array2.push_back(array[j]);
        }
        sort(array2.begin(), array2.end());
        answer.push_back(array2[commands[i][2] - 1]);
    }
    return answer;
}

(2) Python

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