본문 바로가기

코딩테스트/프로그래머스

[Level 1][C++] K번째 수

#include <string>
#include <vector>
#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> test;
        for(int j=commands[i][0]-1; j<commands[i][1] ; j++)
            test.push_back(array[j]);
        sort(test.begin(), test.end());
        answer.push_back(test[commands[i][2]-1]);
    }
    
    return answer;
}