실패율 C++

문제 설명

 

제한사항

• 스테이지의 개수 N은 1 이상 500 이하의 자연수이다.
• stages의 길이는 1 이상 200,000 이하이다.
• stages에는 1 이상 N + 1 이하의 자연수가 담겨있다.
  • 각 자연수는 사용자가 현재 도전 중인 스테이지의 번호를 나타낸다.
  • 단, N + 1 은 마지막 스테이지(N 번째 스테이지) 까지 클리어 한 사용자를 나타낸다.
• 만약 실패율이 같은 스테이지가 있다면 작은 번호의 스테이지가 먼저 오도록 하면 된다.
• 스테이지에 도달한 유저가 없는 경우 해당 스테이지의 실패율은 0 으로 정의한다.

입출력 예

N                    stages                                                                                                          result

5	[2, 1, 2, 6, 2, 4, 3, 3]	[3,4,2,1,5]
4	[4,4,4,4,4]	[4,1,2,3]

입출력 예 설명

입출력 예 #1
1번 스테이지에는 총 8명의 사용자가 도전했으며, 이 중 1명의 사용자가 아직 클리어하지 못했다. 따라서 1번 스테이지의 실패율은 다음과 같다.

• 1 번 스테이지 실패율 : 1/8

2번 스테이지에는 총 7명의 사용자가 도전했으며, 이 중 3명의 사용자가 아직 클리어하지 못했다. 따라서 2번 스테이지의 실패율은 다음과 같다.

• 2 번 스테이지 실패율 : 3/7

마찬가지로 나머지 스테이지의 실패율은 다음과 같다.

• 3 번 스테이지 실패율 : 2/4
• 4번 스테이지 실패율 : 1/2
• 5번 스테이지 실패율 : 0/1

각 스테이지의 번호를 실패율의 내림차순으로 정렬하면 다음과 같다.

• [3,4,2,1,5]

입출력 예 #2

모든 사용자가 마지막 스테이지에 있으므로 4번 스테이지의 실패율은 1이며 나머지 스테이지의 실패율은 0이다.

• [4,1,2,3]

 

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

using namespace std;

bool cmp (const pair<int,double> &a, const pair<int,double> &b)
{
    if(a.second==b.second) 
    {return a.first<b.first;}
    else 
    {return a.second>b.second;}
}

vector<int> solution(int N, vector<int> stages) {
    vector<int> answer;
    vector<pair<int, double>> progress2; //인덱스, 실패율
    sort(stages.begin(), stages.end());
    map<int, int> stagesClear;    //인덱스, 깬 사람수
    int clearPeople = stages.size(); //깨지 못한 사람들 뺄 예정
    for(int i=0; i<stages.size(); i++)
    {
        stagesClear[stages[i]]++;
    }
    
    for(int i=1; i<=N; i++)
    {
        if(clearPeople == 0 || stagesClear[i] == 0)//0인경우
        {
          progress2.push_back({i,0});
        }
        else
        {
              progress2.push_back({i,(double)stagesClear[i] / clearPeople});
        }
        clearPeople -=stagesClear[i];
    }
    
    sort(progress2.begin(), progress2.end(), cmp);
    
    for(int i=0; i<progress2.size(); i++)
    {
        answer.push_back(progress2[i].first);
    }
    return answer;
}