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

프로그래머스 코딩 테스트 : PCCE 기출 문제(C++)

https://school.programmers.co.kr/learn/challenges?order=recent&page=1&languages=cpp&partIds=56388

코딩테스트 연습 | 프로그래머스 스쿨

---

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

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

프로그래머스

---

문제 설명

AI 엔지니어인 현식이는 데이터를 분석하는 작업을 진행하고 있습니다.

데이터는 ["코드 번호(code)", "제조일(date)", "최대 수량(maximum)", "현재 수량(remain)"]으로 구성되어 있으며 현식이는 이 데이터들 중 조건을 만족하는 데이터만 뽑아서 정렬하려 합니다.

예를 들어 다음과 같이 데이터가 주어진다면

data = [ [1, 20300104, 100, 80], [2, 20300804, 847, 37], [3, 20300401, 10, 8] ]

이 데이터는 다음 표처럼 나타낼 수 있습니다.

code	date	maximum	remain
1	20300104	100	80
2	20300804	847	37
3	20300401	10	8

주어진 데이터 중 "제조일이 20300501 이전인 물건들을 현재 수량이 적은 순서"로 정렬해야 한다면 조건에 맞게 가공된 데이터는 다음과 같습니다.

data = [ [3,20300401,10,8],[1,20300104,100,80] ]

정렬한 데이터들이 담긴 이차원 정수 리스트 data와 어떤 정보를 기준으로 데이터를 뽑아낼지를 의미하는 문자열 ext, 뽑아낼 정보의 기준값을 나타내는 정수 val_ext, 정보를 정렬할 기준이 되는 문자열 sort_by가 주어집니다.

data에서 ext 값이 val_ext보다 작은 데이터만 뽑은 후, sort_by에 해당하는 값을 기준으로 오름차순으로 정렬하여 return 하도록 solution 함수를 완성해 주세요.

단, 조건을 만족하는 데이터는 항상 한 개 이상 존재합니다.

---

제한사항

• 1 ≤ data의 길이 ≤ 500
  • data[i]의 원소는 [코드 번호(code), 제조일(date), 최대 수량(maximum), 현재 수량(remain)] 형태입니다.
  • 1 ≤ 코드 번호≤ 100,000
  • 20000101 ≤ 제조일≤ 29991231
  • data[i][1]은 yyyymmdd 형태의 값을 가지며, 올바른 날짜만 주어집니다. (yyyy : 연도, mm : 월, dd : 일)
  • 1 ≤ 최대 수량≤ 10,000
  • 1 ≤ 현재 수량≤ 최대 수량
• ext와 sort_by의 값은 다음 중 한 가지를 가집니다.
  • "code", "date", "maximum", "remain"
  • 순서대로 코드 번호, 제조일, 최대 수량, 현재 수량을 의미합니다.
• val_ext는 ext에 따라 올바른 범위의 숫자로 주어집니다.
• 정렬 기준에 해당하는 값이 서로 같은 경우는 없습니다.

---

입출력 예

입출력 예

입출력 예 설명

---

코드

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

using namespace std;

int sort_num;

int str_to_num(string str) {
    vector<string> v = { "code", "date", "maximum", "remain" };
    for (int index = 0; index < 4; index++)
        if (v[index] == str)
            return index;
}

bool compare(vector<int> v1, vector<int> v2) {
    return v1[sort_num] < v2[sort_num];
}

vector<vector<int>> solution(vector<vector<int>> data, string ext, int val_ext, string sort_by) {
    vector<vector<int>> answer;
    int ext_num = str_to_num(ext);
    sort_num = str_to_num(sort_by);
    
    for (auto arr : data)
        if (arr[ext_num] < val_ext)
            answer.push_back(arr);
    
    sort(answer.begin(), answer.end(), compare);
    
    return answer;
}

// 다른 사람의 풀이 : 함수를 따로 만들지 않고 해결하는 방법

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

using namespace std;

vector<vector<int>> solution(vector<vector<int>> data, string ext, int val_ext, string sort_by) {
    vector<vector<int>> answer;
    unordered_map<string, int> name_to_idx;
    name_to_idx["code"] = 0;
    name_to_idx["date"] = 1;
    name_to_idx["maximum"] = 2;
    name_to_idx["remain"] = 3;

    auto it = remove_if(data.begin(), data.end(), [idx=name_to_idx[ext], val_ext](const auto& row) {
        return !(row[idx] < val_ext);
    });
    data.erase(it, data.end());

    sort(data.begin(), data.end(), [idx=name_to_idx[sort_by]](const auto& row1, const auto& row2) {
        return row1[idx] < row2[idx];
    });
    return data;
}

---

결과

---

여담

이번 문제도 크게 어렵진 않았지만 함수(str_to_num)를 만들지 말고 map을 사용했으면 더 간결했을텐데 생각 못한게 조금 아쉽긴 하다.

1레벨은 0레벨에 비해 문제 난이도 자체는 크게 어렵지는 않지만, 따로 함수를 만들거나 기존에 배운걸 잘 응용해야 하는 문제 위주로 진행되는 것같다.