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

문제

---

문제를 보시려면 링크를 클릭해주세요.

풀이

---

1. 풀이 1,2 모두 tmp 벡터에 array의 i~j번째까지 자른 값을 담아줍니다.

   풀이 2의 경우 벡터의 복사 생성자를 이용하여 슬라이싱

2. sort 함수로 정렬합니다.

3. tmp 벡터의 k번째 수를 answer 벡터에 담아줍니다.

소스 코드

---

풀이 1

#include <algorithm>
#include <vector>

using namespace std;

vector<int> solution(vector<int> array, vector<vector<int>> commands) {
    vector<int> answer;
    
    for(vector<int> v:commands){
        vector<int> tmp;
        for(int j=v[0]-1; j<v[1]; j++){
            tmp.push_back(array[j]);
        }
        sort(tmp.begin(), tmp.end());
        answer.push_back(tmp[v[2]-1]);
    }
    return answer;
}

풀이 2

#include <algorithm>
#include <vector>

using namespace std;

vector<int> solution(vector<int> array, vector<vector<int>> commands) {
    vector<int> answer;
    
    for(vector<int> v:commands){
        vector<int> tmp = vector<int>(array.begin()+v[0]-1, array.begin()+v[1]);
        sort(tmp.begin(), tmp.end());
        answer.push_back(tmp[v[2]-1]);
    }
    return answer;
}