프로그래머스 - lv1 K번째 수

K번째 수

문제는 프로그래머스에서 확인 할 수 있다.

✔ 접근방법

Vector를 이용하면 쉽게 문제를 해결할 수 있다.

1. 주어진 인덱스에 해당하는 새로운 벡터를 만든다.
2. 정렬한다.

✔ 코드

#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;

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

    return answer;
}

int main(void){

    vector<int> array = {1,5,2,6,3,7,4};
    vector<vector<int>> commands = {{2,5,3},{4,4,1},{1,7,3}};
    vector<int> ret;

    ret = solution(array, commands);

    return 0;
}

☝ 팁

• sort() 함수를 통해 정렬을 수행할 수 있다. 이때 인자로는 iterator를 넘겨준다.

👍 참고 사이트

프로그래머스