[프로그래머스/C++] PCCE 기출문제 10번 - 데이터 분석 - (1 swap, 변수 줄이기)

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

입출력 예

고칠 점 1 코드 간결화

여러 데이터 제목 중 한 가지를 선택한 뒤, 그 안에 도는 코드가 똑같다면 미리 하나의 변수를 입력받게 설정하고 통일해서 다음 코드로 들어가게 만들어라

 if (sort_by == "code")
     num = 0;
 else if (sort_by == "date")
     num = 1;
 else if (sort_by == "maximum")
     num = 2;
 else
     num = 3;

미리 num을 설정해놓으면 뒤에 for문을 돌릴 때 시간, 메모리, 코드 간결성 모든 면에서 뛰어나진다.

 

고칠 점 2 swap 함수 사용하기

사실 밑에 코드 부분은 swap을 하면 훨씬, 간결하게 사용할 수 있다.

코드 간결성이 무너지니 중간에 조그만한 실수가 발생하게 되고 이는 결국 시간을 잡아먹는 요인이 된다. 

vector<vector<int>> tmp;
if (answer[j][num] > answer[i][num]) {
    tmp.push_back(answer[i]);
    answer[i] = answer[j];
    answer[j] = tmp[0];
}

내 코드 

#include <string>
#include <vector>
#include <algorithm>
using namespace std;

vector<vector<int>> solution(vector<vector<int>> data, string ext, int val_ext, string sort_by) {
    vector<vector<int>> answer;
    for (int i = 0; i < data.size(); i++) {
        if (ext == "code") {
            if (data[i][0] < val_ext) answer.push_back(data[i]);
        }
        else if (ext == "date") {
            if (data[i][1] < val_ext) answer.push_back(data[i]);
        }
        else if (ext == "maximum") {
            if (data[i][2] < val_ext) answer.push_back(data[i]);
        }
        else if (ext == "remain") {
            if (data[i][3] < val_ext) answer.push_back(data[i]);
        }
    }
    int num;
if (sort_by == "code")
    num = 0;
else if (sort_by == "date")
    num = 1;
else if (sort_by == "maximum")
    num = 2;
else
    num = 3;
for (int j = 0; j < answer.size() - 1; j++) {
    for (int i = j + 1; i < answer.size(); i++) {

        vector<vector<int>> tmp;
        if (answer[j][num] > answer[i][num]) {
            tmp.push_back(answer[i]);
            answer[i] = answer[j];
            answer[j] = tmp[0];
        }
    


        }
    }
    return answer;
}

 

최적화 코드 

#include <string>
#include <vector>

using namespace std;

vector<vector<int>> solution(vector<vector<int>> data, string ext, int val_ext, string sort_by) {
    vector<vector<int>> answer;
    int num = 0;
    if (ext == "code")
        num = 0;
    else if (ext == "date")
        num = 1;
    else if (ext == "maximum")
        num = 2;
    else
        num = 3;
    for (int i = 0; i < data.size(); ++i)
    {
        if (data[i][num] < val_ext)
            answer.push_back(data[i]);
    }
    if (sort_by == "code")
        num = 0;
    else if (sort_by == "date")
        num = 1;
    else if (sort_by == "maximum")
        num = 2;
    else
        num = 3;
    for (int i = 0; i < answer.size() - 1; ++i)
    {
        for (int j = i + 1; j < answer.size(); ++j)
        {
            if (answer[i][num] > answer[j][num])
                swap(answer[i], answer[j]);
        }
    }
    return answer;
}