[Algorithm] 프로그래머스 C++ : K번째수

 

 

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

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

 

Level1 문제는 풀 때마다 즐겁다.. 빨리 풀려서.. (짱쉬우니까..)

 

암튼 그냥 문제 condition에 맞춰서

 주어진 commands 명령들을 반복문으로 하나씩 돌면서 answer에 답을 추가해주면 된다.

 

벡터 index만 잘 맞춰서 리턴하면 되는 문제

 

 

Code

#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 cmd: commands) {
        int start = cmd[0], end = cmd[1], k = cmd[2];
        
        vector<int> tmp = array;
        sort(tmp.begin()+start-1, tmp.begin()+end);
        
        answer.push_back(tmp[start+k-2]);
    }
    
    return answer;
}