프로그래머스 (실패율, 2019 KAKAO BLIND RECRUITMENT) C++

 

 카카오 문제로 vector가 주어지는 경우 거의 이중 for문으로 시작해야 하는 문제가 많은 것 같다. 오랜만에 주석을 먼저 달고 코드를 짰는데 생각보다 잘 짜졌다. Map을 활용해서 value에 대해 오름차순/내림차순 정렬하는 법을 체화한 것 같다.

 

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

using namespace std;


bool cmp(const pair<int, double>& a, const pair<int, double>& b) {
    // value가 같다면 key에 해당하는 것은 오름차순으로 정렬한다.
    if(a.second == b.second) return a.first < b.first;
    
    // value에 대해 내림차순으로 정렬한다.
    else return a.second > b.second;
}

vector<int> solution(int N, vector<int> stages) {
    vector<int> answer;
    map<int, double> fail_ratio;
    int challenger = stages.size();
    int failer = 0;
    
    for(int j = 1 ; j <= N ; j++) {
        for(int i = 1 ; i <= stages.size() ;i++) {
            // 해당 stage를 만나면 몇 명이 도전을 했고 몇 명이 실패했는지 체크한다.
            if(stages[i-1] == j) 
                failer++;
        }
        
        // 모든 도전자가 stage가 끝나기 전에 없어지는 경우 0으로 나눌 수 없으므로
        if(challenger == 0)
            fail_ratio[j] = 0;
        else
            // map 변수에 key는 stage, value는 실패율을 담아준다.
            fail_ratio[j] = (double) failer / (double) challenger;

        challenger -= failer;
        failer = 0;
    }
    
    // map의 value에 해당하는 것을 내림차순으로 정렬하기 위한 알고리즘
    vector<pair<int,double>> temp (fail_ratio.begin(), fail_ratio.end());
    sort(temp.begin(), temp.end(), cmp);
    
    for(auto cnt : temp){
        cout << cnt.first << " " << cnt.second << endl;
        answer.push_back(cnt.first);
    }
    
    return answer;
}