[프로그래머스] [PCCE 기출문제] 10번 / 데이터 분석

---

🔗 문제 링크

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

프로그래머스

---

👩‍💻 코드

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

using namespace std;

int index_sort;

bool cmp(const vector<int>& a, const vector<int>& b) {
    return a[index_sort] < b[index_sort];
}

vector<vector<int>> solution(vector<vector<int>> data, string ext, int val_ext, string sort_by) {
    vector<vector<int>> answer;
    vector<string> num_ext = { "code", "date", "maximum", "remain" };
    int index_ext;
    
    for(int i=0; i<num_ext.size(); i++){
        if(ext == num_ext[i]){
            index_ext = i;
        }
        
        if(sort_by == num_ext[i]){
            index_sort = i;
        }
    }
    
    for(vector<int> v : data){
        if(v[index_ext] < val_ext){
            answer.push_back(v);
        }
    }
    
    sort(answer.begin(), answer.end(), cmp);
    
    return answer;
}

---

📝 풀이

맨 처음 for문으로 ext와 sort_by의 인덱스를 구합니다.

sort_by의 인덱스는 비교 함수에서 사용하기 위해 전역 변수로 선언했습니다.

두번째 for문에서는 문제의 첫 조건인 val_ext보다 작은 vector<int>만 뽑아서 answer에 추가합니다.

마지막으로 sort_by 기준의 오름차순으로 정렬하는 cmp 함수를 작성합니다.

아까 구해둔 sort_by의 인덱스를 이용하여 한 줄로 작성합니다.