[프로그래머스/C++ 문제 풀이] Lv. 1 - [PCCE 기출문제] 10번 / 데이터 분석

문제 설명

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

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

data = [[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에 따라 올바른 범위의 숫자로 주어집니다.
• 정렬 기준에 해당하는 값이 서로 같은 경우는 없습니다.

입출력 예

data	ext	val_ext	sort_by	result
[[1, 20300104, 100, 80], [2, 20300804, 847, 37], [3, 20300401, 10, 8]]	"date"	20300501	"remain"	[[3,20300401,10,8],[1,20300104,100,80]]

 

입출력 예 #1

• 본문의 내용과 동일합니다.

---

문제 풀이

이번 문제는 1차로 주어진 정보(string)에 따라 데이터를 추출하고, 해당 데이터를 바탕으로 2차 정렬(sort)시켜야 하는 문제입니다.

 

저의 경우 <string(정보), int(인덱스)> 형식의 map 컨테이너를 사용해 string으로 주어진 정보를 실제 인덱스에 맞게 매핑시켰으며, sort() 함수에 비교 함수(Comp)를 직접 작성해 추가해주는 형식으로 문제를 해결하게 되었습니다.

정답 코드

풀이 시간 : 13m 22s

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

using namespace std;

// 정렬할 문자열 cache
string sort_by_str;
// 문자열에 따른 정보 인덱스 map
map<string, int> m;

// 비교 함수
bool Comp(const vector<int>& v1,const vector<int>& v2)
{
    return v1[m[sort_by_str]] < v2[m[sort_by_str]];
}

vector<vector<int>> solution(vector<vector<int>> data, string ext, int val_ext, string sort_by) {
    vector<vector<int>> answer;

    // 정보 매핑
    m.insert(make_pair("code", 0));
    m.insert(make_pair("date", 1));
    m.insert(make_pair("maximum", 2));
    m.insert(make_pair("remain", 3));

    // ect 조건에 맞는 데이터 추출
    for (int i = 0; i < data.size(); i++)
    {
        // 조건에 맞는 값보다 실제 값이 작은 경우
        if (data[i][m[ext]] < val_ext)
        {
            // 결과에 추가
            answer.push_back(data[i]);
        }
    }

    sort_by_str = sort_by;

    // 정렬
    sort(answer.begin(), answer.end(), Comp);


    return answer;
}

sort() 함수에 사용자 정의 비교 함수를 3번째 인자로 넣을 수 있다는 정보만 알면 그리 어려운 문제는 아니라고 생각합니다.

 

https://medium.com/letsps/c-sort%ED%95%A8%EC%88%98%EC%97%90-%EC%82%AC%EC%9A%A9%EC%9E%90-%EC%A0%95%EC%9D%98%ED%95%A8%EC%88%98-operator%EB%A5%BC-%EB%84%A3%EC%96%B4%EB%B3%B4%EC%9E%90-21e920699c83