본문 바로가기
알고리즘/프로그래머스

[프로그래머스] k번째수 c++

by 오오오니 2024. 1. 16.

풀이

주어진 벡터의 일부분을 새로 정렬했을때 특정한 순서에 있는 숫자가 무엇인지 출력하는 문제이다.

일부분에 해당하는 범위를 새로운 벡터에 할당하고 정렬한다음에 특정한 순서에 있는 숫자를 answer에 담았다.

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