[BOJ/C++] 올림픽(8979번)

1. 개요

https://www.acmicpc.net/problem/8979

8979번: 올림픽

1-1. 설명

올림픽 등수를 정하기 위해 아래의 조건들을 통해 등수를 정한다. 

 - 금메달 수가 더 많은 나라

 - 금메달 수 동일 시 은메달 수가 많은 나라

 - 은메달 수 동일 시 동매달 수가 많은 나라

각 국가는 1~N사이 정수로 표현되며, 한 국가의 등수는 (자신보다 더 잘한 나라 수) + 1 로 정의된다. 만약 두 나라가 금, 은, 동메달 수 동일할 시 두 나라의 등수는 같다. 각 국가의 금, 은, 동 정보를 받아 어느 국가가 몇등했는지 알려주는 프로그램을 구현한다. 

1-2. 제한 사항

 - 입력 첫 줄은 국가의 수 N, 등수를 알고싶은 국가K가 주어짐

 - N은 1 이상 1,000 이하, K는 1 이상 N 이하

 - 각 국가는 1부터 N사이 정수로 표현됨

---

2. 구현

2-1. 풀이

금, 은, 동, 국가 번호를 저장하는 구조체를 생성하고 이를 배열에 저장한다. sort를 통해 내림 차순으로 정렬한 후, 메달 개수가 동일할 땐 rank 유지, 다를 땐 i만큼 더해 rank에 저장한다

2-2. 코드

#include <iostream>
#include <algorithm>
using namespace std;

struct Medal {
    int gold;
    int silver;
    int bronze;
    int index;
};

constexpr size_t MAX_NUM = 1001;
Medal nums[MAX_NUM];

bool compare(Medal a, Medal b) {
    if (a.gold == b.gold) {
        if (a.silver == b.silver) return a.bronze > b.bronze;
        else return a.silver > b.silver;
    }
    else return a.gold > b.gold;
}

int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    int N, K; // 국가의 수 , 등수를 알고싶은 국가
    int gold, silver, bronze, index;
    cin >> N >> K;
    for (int i = 0; i < N; i++) {
        cin >> index >> gold >> silver >> bronze;
        nums[i] = { gold, silver, bronze, index };
    }
    sort(nums, nums + N, compare);

    int rank = 1;
    for (int i = 0; i < N; i++) {
        if ((nums[i - 1].gold != nums[i].gold) || (nums[i - 1].silver != nums[i].silver) || (nums[i - 1].bronze != nums[i].bronze)) rank = i + 1;
        if (nums[i].index == K) break;
    }
    cout << rank << "\n";
    return 0;
}