프로그래머스 Level 1 데이터 분석

 

 

---

 

코드

#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;
    
    int index=0;
    
    if(ext=="code")
    {
        index=0;
    }else if(ext=="date")
    {
        index=1;
    }else if(ext=="maximum")
    {
        index=2;
    }else if(ext=="remain")
    {
        index=3;
    }
    
    for(int i=0;i<data.size();i++)
    {
        if(val_ext>data[i][index])
        {
            answer.push_back(data[i]);
        }
    }
    
    if(sort_by=="code")
    {
        index=0;
    }else if(sort_by=="date")
    {
        index=1;
    }else if(sort_by=="maximum")
    {
        index=2;
    }else if(sort_by=="remain")
    {
        index=3;
    }
    
    if(answer.size()==1)
    {
        return answer;
    }
        
    // 오름차순 정렬
    vector<int> temp;
    
    for(int i=0;i<answer.size()-1;i++){
        for(int j=i+1;j<answer.size();j++){
            if(answer[i][index]>answer[j][index]){
                temp=answer[i];
                answer[i]=answer[j];
                answer[j]=temp;                
            }
        }
    }
    
    return answer;
}