자세한 설명은 모두 주석으로 달아놓았다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
#include <string>
#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;
bool comp(const pair<double,int>& a,const pair<double,int>& b){
if(a.first==b.first){ //first가 같다면 스테이지 번호가 작은 순(오름차순)
return a.second<b.second;
}
return a.first>b.first; //아니라면 내림차순 정렬
}
vector<int> solution(int N, vector<int> stages) {
vector<int> answer;
//스테이지에 도달했으나 아직 클리어하지 못한 플레이어 수와
//스테이지에 도달한 플레이어 수 저장할 vector
vector<pair<int,int>> v;
//실패율, 스테이지 N번째 저장할 vector
vector<pair<double,int>> rate;
for(int i=1;i<=N;i++){
int ncnt=0, ocnt=0; //도달했는데 아직 클리어 못함/도달한 수
for(int j=0;j<stages.size();j++){
if(stages[j]==i) ncnt++;
if(stages[j]>=i) ocnt++;
}
if(ocnt==0) v.push_back({0,0}); //스테이지에 도달한 유저가 없는 경우
else v.push_back({ncnt,ocnt});
}
for(int i=0;i<N;i++){
if(v[i].first==0) {rate.push_back({0,i+1}); continue;} //0이면 못나눔.
double tmp = (double)v[i].first/v[i].second; //실패율 계산
rate.push_back({tmp,i+1});
}
sort(rate.begin(),rate.end(),comp); //내림차순 정렬
for(int i=0;i<N;i++){
answer.push_back(rate[i].second);
}
return answer;
}
|
cs |