[알고리즘] K번째 수

포인트

1. stl에 사용하는 것에 익숙해질 필요가 있다. 

2. 정렬은 (sort, stable_sort 가 있다.)

🧶문서는 항상 수정 될 수 있습니다. 비판은 환영합니다. 

c++/cpp

#include <string>
#include <vector>
#include <algorithm>
#include <iostream>

using namespace std;

vector<int> solution(vector<int> array, vector<vector<int>> commands) {
    vector<int> answer;

    for (auto ele : commands) {
        vector<int> vc;
        vc.assign(array.begin() + ele[0] - 1, array.begin() + ele[1]);
        sort(vc.begin(), vc.end());
        answer.push_back(vc[ele[2] - 1]);
    }

    return answer;
}

python 

def solution(array, commands):
    answer = []
    for ele in commands:
        temp = sorted(list(array[ele[0]-1:ele[1]]))
        answer.append(temp[ele[2]-1])
    return answer

결국은 간단한 정렬을 문제의 조건에 따라서 구할 수 있는가를 물어보는 문제이다.