K번째수

문제

풀이

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

using namespace std;

vector<int> solution(vector<int> array, vector<vector<int>> commands) {
    vector<int> answer;
    vector<int> v;
    cout<<commands.size();
    for(int i=0;i<commands.size();i++)
    {

        int x=commands[i][0];
        int y=commands[i][1];
        int z=commands[i][2];
        
        for(int j=x-1; j < y ;j++)
        {
            v.push_back(array[j]);
        }
        
        sort(v.begin(),v.end());
        answer.push_back(v[z-1]);
        
        v.clear();
    }
    return answer;
}

vector를 하나 선언하여 각각에 쪼개어진 데이터를 추가하고 sort를 하였다. 그리고 answer에서 추출해야 하는 값을 넣어주고 vector를 초기화해주었다.