안녕 세상아,

[프로그래머스/c++] Lv1 K번째수 본문

프로그래머스

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

돈 많은 백수가 되고싶다 2024. 9. 13. 16:19

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

2차원 배열이든 벡터든 너무 어려워...ㅠ

 

1. 새로운 v 벡터를 만든 후 i번째 숫자부터 j번째 숫자까지를 벡터에 넣는다. 

2. sort 정렬을 한다. 

3. v 벡터에서 k 번째 수를 answer 벡터에 넣는다. 

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