[프로그래머스] #42889 실패율

2018년 KAKAO BLIND RECRUITMENT

코딩테스트 연습 - 실패율 | 프로그래머스

 

문제

실패율을 구하는 코드를 완성하라.

• 실패율은 다음과 같이 정의한다.
  • 스테이지에 도달했으나 아직 클리어하지 못한 플레이어의 수 / 스테이지에 도달한 플레이어 수

전체 스테이지의 개수 N, 게임을 이용하는 사용자가 현재 멈춰있는 스테이지의 번호가 담긴 배열 stages가 매개변수로 주어질 때, 실패율이 높은 스테이지부터 내림차순으로 스테이지의 번호가 담겨있는 배열을 return 하도록 solution 함수를 완성하라.

풀이

제한사항을 제대로 못 보고 풀어서 틀렸었다.

스테이지에 도달한 유저가 없는 경우 해당 스테이지의 실패율은 0 으로 정의한다.

이 부분이 있는데 틀렸을 때의 로직은 도달한 스테이지 번호들 중 가장 큰 것을 Max라고 두고 v 라는 vector 객체에 넣을 때 Max부터 순회하도록 했었다.

하지만 이 부분은 예제가 아래와 같이 나와 있을 때는

#testcase
7
1,2,3,4,5,6

예상 답안이 [6,5,4,3,2,1,7] 이지만 7이 출력되지 않는 현상이 발생한다. 따라서 Max를 N으로 초기화 시켜두고 반복문을 돌 때 Max보다 큰 스테이지 번호가 나오면 갱신시키 도록 수정하니 맞출수 있었다.

 

코드

#include <algorithm>
#include <vector>
using namespace std;
typedef pair<double, int> p;
 
bool cmp(p a, p b) {
    if (a.first == b.first) {
        return a.second < b.second;
    }
    return a.first > b.first;
}
 
vector<int> solution(int N, vector<int> stages) {
    vector<int> answer;
    vector<p> v;
    int arr[502] = { 0 }, Max = N;
    for (int i = 0; i < stages.size(); i++) {
        arr[stages[i]]++;
        Max = stages[i] > Max ? stages[i] : Max;
    }
    int people = 0;
    for (int i = Max; i > 0; i--) {
        people += arr[i];
        if (i <= N) {
            if (!arr[i])v.push_back({ 0, i });
            else v.push_back({ ((double)arr[i] / people), i });
        }
    }
    sort(v.begin(), v.end(), cmp);
    for (int i = 0; i < v.size(); i++) {
        answer.push_back(v[i].second);
    }
    return answer;
}
 
int main() {
    solution(5, { 2, 1, 2, 6, 2, 4, 3, 3 });
    return 0;
 
}