[프로그래머스] K번째수

programmers.co.kr/learn/courses/30/lessons/42748

코딩테스트 연습 - K번째수

---

문제풀이)

이건 벡터 사용법을 익히는 문제같다. 해당 commands에 맞는 수행을 한 뒤, sort함수를 이용하여 정렬하고, 정렬한 배열에서 k번째 수를 answer에 push_back해주면 된다.

 

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