백준 8979번 올림픽

문제 링크입니다: https://www.acmicpc.net/problem/8979

정렬(Sorting) 알고리즘 문제였습니다.

알고리즘은 아래와 같습니다.

1. pair<pair<int, int>, pair<int, int>>형 배열을 선언하여 국가번호, 금, 은, 동 순으로 입력받습니다.

2. 정렬 조건을 위해 cmp 함수를 작성하는데 아래와 같이 작성합니다.

i) 금을 기준으로 내림차순 정렬

ii) 금이 같을 때 은을 기준으로 내림차순 정렬

iii) 금, 은이 같을 때 동을 기준으로 내림차순 정렬

iv) 금, 은, 동이 같을 때 K번째 나라가 제일 앞에 오도록

3. 배열을 탐색하면서 국가번호가 K번째인 나라의 인덱스를 출력해주면 됩니다.

#include <iostream>

#include <algorithm>

using namespace std;

 

const int MAX = 1000 + 1;

 

int N, K;

//idx, gold, silver, bronze

pair<pair<int, int>, pair<int, int>> medal[MAX];

 

bool cmp(pair<pair<int, int>, pair<int, int>> a, pair<pair<int, int>, pair<int, int>> b)

{

        //금

        if (a.first.second > b.first.second)

                 return true;

        else if (a.first.second == b.first.second)

        {

                 //은

                 if (a.second.first > b.second.first)

                         return true;

                 else if (a.second.first == b.second.first)

                         //동

                         if (a.second.second > b.second.second)

                                 return true;

                         //동일 랭크에 대해 K번째 나라가 제일 먼저 오도록

                         else if (a.second.second == b.second.second)

                                 return a.first.first == K;

        }

        return false;

}

 

int main(void)

{

        ios_base::sync_with_stdio(0);

        cin.tie(0); //cin 실행속도 향상

        cin >> N >> K;

 

        for (int i = 0; i < N; i++)

                 cin >> medal[i].first.first >> medal[i].first.second >> medal[i].second.first >> medal[i].second.second;

 

        sort(medal, medal + N, cmp);

 

        for (int i = 0; i < N; i++)

                 if (medal[i].first.first == K)

                 {

                         cout << i + 1 << "\n";

                         break;

                 }

        return 0;

}

개발환경:Visual Studio 2017

지적, 조언, 질문 환영입니다! 댓글 남겨주세요~