[프로그래머스] 실패율(C++)

문제 링크

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

프로그래머스

 

 

코드 구현

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

using namespace std;

bool compare(const pair<int, float>& a, const pair<int, float>& b)
{
    if (a.second == b.second)
        return a.first < b.first;

    return a.second > b.second;
}

vector<int> solution(int N, vector<int> stages)
{
    vector<int> answer;
    vector<float> challenger(N + 2, 0.0);
    vector<float> fail(N + 2, 0.0);

    // vector<int> stages = { 2, 1, 2, 6, 2, 4, 3, 3 }; 
    for (int i = 0; i < stages.size(); i++)
    {
        for (int j = 1; j <= stages[i]; j++)
            challenger[j]++;

        fail[stages[i]]++;
    }

    vector<pair<int, float>> failRatio(N);

    for (int i = 0; i < N; i++)
    {
        float failRate = fail[i + 1] / challenger[i + 1];

        failRatio[i].first = i + 1;

        if (challenger[i + 1] == 0)
            failRatio[i].second = 0;
        else
            failRatio[i].second = failRate;
    }

    sort(failRatio.begin(), failRatio.end(), compare);

    for (int i = 0; i < N; i++)
        answer.push_back(failRatio[i].first);

    return answer;
}

 

 

해결 방법

스테이지에 도달한 플레이어의 수와 도달했지만 클리어하지 못한 플레이어의 수를 각각 벡터에 구한 후, pair<int, float> 을 담고있는 벡터를 이용하여 <스테이지, 실패율> 로 저장한다. 

따로 구현한 compare 함수를 이용하여 정렬을 하고 스테이지를 반환하였다.