프로그래머스 - K번째수 - C++

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

프로그래머스

 

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

    return answer;
}

 

설명

이 문제는 주어진 커맨드를 가지고 배열을 잘라서 그 배열 중 k번째 수를 골라내는 문제이다. 

 

아주 단순하게 풀면된다. 

바깥 for문으로 커맨드의 수만큼 돌면서 안쪽 for문으로 [2,5,3]이면

array의 2번째에서 5번째 인덱스까지를 a 벡터에 넣고 sort함수로 오름차순 정렬을 한다.

그리고 a 벡터에서 3번째수를 answer에 넣어주면 된다.