알고리즘/BOJ

백준 8979번 올림픽

꾸준함. 2018. 8. 5. 02:20

문제 링크입니다: 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


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

반응형

'알고리즘 > BOJ' 카테고리의 다른 글

백준 1717번 집합의 표현  (0) 2018.08.05
백준 15961번 회전 초밥  (0) 2018.08.05
백준 8980번 택배  (0) 2018.08.04
백준 1449번 수리공 항승  (0) 2018.08.04
백준 1969번 DNA  (0) 2018.08.04