코딩테스트 준비/프로그래머스

[프로그래머스] - k번째 수(정렬) c++

SeoburiFaust 2024. 2. 14. 20:21

문제

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

 

프로그래머스

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

programmers.co.kr

접근방법

굳이 lamda를 한 번 써봤다. 

코드

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

vector<int> solution(vector<int> array, vector<vector<int>> commands) {
    vector<int> answer;
    int n = commands.size();
    for (int tc=0;tc<n;tc++) {
        int i = commands[tc][0] - 1;
        int j = commands[tc][1] - 1;
        int k = commands[tc][2] - 1;
        vector<int> temp;
        for (int p = i; p <= j; p++) {
            temp.push_back(array[p]);
        }
        sort(temp.begin(), temp.end(), [](int a, int b) {
            return a < b;
        });
        answer.push_back(temp[k]);
    }
    
    return answer;
}

개선할 점

sort할 때 lamda에 더 익숙해지면 좋을 거 같다.