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

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

프로그래머스

<풀이>

1. 주어진 입력으로 실패율을 구하는데 필요한 분모와 분자를 구한다.

2. 스테이지의 번호와 실패율을 벡터에 저장한다.

3. 주어진 조건에 따라 벡터를 정렬한다.

4. 결과를 저장한다.

#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
using namespace std;
 
bool cmp(pair<int,double> a, 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;
 
    int bunJa[502];
    int bunMo[502];
    memset(bunJa, 0, sizeof(bunJa));
    memset(bunMo, 0, sizeof(bunMo));
 
    //vector<스테이지 번호, 실패율>
    vector<pair<int,double>> rate;
 
    //각각 스테이지 실패율의 분모, 분자 구하기
    for (int i = 0; i < stages.size(); i++) {
        for (int j = 1; j <= stages[i]; j++) {
            bunMo[j]++;
        }
        bunJa[stages[i]]++;
    }
 
    //실패율 구하기
    for (int i = 1; i <= N; i++) {
        if (bunMo[i] == 0) {
            rate.push_back({ i,0 });
        }
        else {
            rate.push_back({ i,double(bunJa[i]) / double(bunMo[i]) });
        }
    }
 
    //실패율 정렬 
    sort(rate.begin(), rate.end(), cmp);
 
    //결과 저장
    for (int i = 0; i < N; i++) {
        answer.push_back(rate[i].first);
    }
 
    return answer;
}

 

구현에 대해 알아볼 수 있는 문제였습니다.