백준 8979번 풀이 c++ and java

문제

올림픽은 참가에 의의가 있기에 공식적으로는 국가간 순위를 정하지 않는다. 그러나, 많은 사람들이 자신의 국가가 얼마나 잘 하는지에 관심이 많기 때문에 비공식적으로는 국가간 순위를 정하고 있다. 두 나라가 각각 얻은 금, 은, 동메달 수가 주어지면, 보통 다음 규칙을 따라 어느 나라가 더 잘했는지 결정한다.

1. 금메달 수가 더 많은 나라 
2. 금메달 수가 같으면, 은메달 수가 더 많은 나라
3. 금, 은메달 수가 모두 같으면, 동메달 수가 더 많은 나라 

각 국가는 1부터 N 사이의 정수로 표현된다. 한 국가의 등수는 (자신보다 더 잘한 나라 수) + 1로 정의된다. 만약 두 나라가 금, 은, 동메달 수가 모두 같다면 두 나라의 등수는 같다. 예를 들어, 1번 국가가 금메달 1개, 은메달 1개를 얻었고, 2번 국가와 3번 국가가 모두 은메달 1개를 얻었으며, 4번 국가는 메달을 얻지 못하였다면, 1번 국가가 1등, 2번 국가와 3번 국가가 공동 2등, 4번 국가가 4등이 된다. 이 경우 3등은 없다. 

각 국가의 금, 은, 동메달 정보를 입력받아서, 어느 국가가 몇 등을 했는지 알려주는 프로그램을 작성하시오. 

입력

입력의 첫 줄은 국가의 수 N(1 ≤ N ≤ 1,000)과 등수를 알고 싶은 국가 K(1 ≤ K ≤ N)가 빈칸을 사이에 두고 주어진다. 각 국가는 1부터 N 사이의 정수로 표현된다. 이후 N개의 각 줄에는 차례대로 각 국가를 나타내는 정수와 이 국가가 얻은 금, 은, 동메달의 수가 빈칸을 사이에 두고 주어진다. 전체 메달 수의 총합은 1,000,000 이하이다.

출력

출력은 단 한 줄이며, 입력받은 국가 K의 등수를 하나의 정수로 출력한다. 등수는 반드시 문제에서 정의된 방식을 따라야 한다. 

서브태스크

번호배점제한

1	8	예제 입력, 출력
2	12	N = 2
3	20	모든 국가의 은메달 및 동메달 획득 수는 0
4	25	N ≤ 500
5	35	추가적인 제약 조건은 없다.

예제 입력 1 복사

4 3
1 1 2 0
2 0 1 0
3 0 1 0
4 0 0 1

예제 출력 1 복사

2

예제 입력 2 복사

4 2
1 3 0 0
3 0 0 2
4 0 2 0
2 0 2 0

예제 출력 2 복사

2

 

 

처음에 이문제를 풀려고 sort를 사용을 했고 금 은 동의 갯수로 내림차순 sort를 하려고 bool comp를 만들어서 비교를 했으나 너무 복잡해져서 다른 해결책을 참고했다. vector<pair<>> 이런식으로 vector를 선언하는 것이 아닌 구조체를 사용해서 gold, silver, bronze, index를 만들고 벡터에 할당해서 사용하는 방법이 하나 있었고 다른 하나는 골드 실버 브론즈 순으로 순차적으로 골드 타겟값 보다 현재 골드 인덱스값이 더 크다면 순위가 내려가는것이므로 res++를 해서 최종적으로는 순위를 구할 수 있는 방법이 있었다. 나는 후자를 선택했다.

 

 

-c++

#include <iostream>
#include <vector>

using namespace std;


int main() {

    int n=0,k=0;

    int gold[1001];
    int silver[1001];
    int bronze[1001];

    cin >> n >> k ;

    for(int i = 1 ;  i<= n; i++){
        int a,b,c,d;
        cin >> a>> b>> c>> d;
        gold[a]=b;
        silver[a]=c;
        bronze[a]=d;
    }

    int res=1;

    for(int i=1; i<=n;i++){

        if(gold[i]>gold[k]){
            res++;
        }else if(gold[i]==gold[k]){
            if(silver[i]>silver[k]){
                res++;
            }else if(silver[i]==silver[k]){
                if(bronze[i]>bronze[k]){
                    res++;
                }
            }
        }
    }

    cout << res <<"\n";







    return 0;
}

 

- java

import java.io.*;
import java.util.*;


public class Main {



    public static void main(String[] args) throws NumberFormatException, IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        StringTokenizer st = new StringTokenizer(br.readLine());

        int n = Integer.parseInt(st.nextToken());
        int k = Integer.parseInt(st.nextToken());


        int [] gold = new int [1001];
        int [] silver = new int [1001];
        int [] bronze = new int [1001];

        for(int i = 0 ; i < n ; i++){
            st = new StringTokenizer(br.readLine());

            int index = Integer.parseInt(st.nextToken());
            int a = Integer.parseInt(st.nextToken());
            int b = Integer.parseInt(st.nextToken());
            int c = Integer.parseInt(st.nextToken());

            gold[index]=a;
            silver[index]=b;
            bronze[index]=c;

        }

        int res = 1;

        for (int i = 1 ; i <= n; i++){

            if(gold[i] > gold[k]){
                res++;
            }else if(gold[i]==gold[k]){
                if(silver[i]>silver[k]){
                    res++;
                }else if(silver[i]==silver[k]){
                    if(bronze[i]>bronze[k]){
                        res++;
                    }
                }
            }
        }

        System.out.println(res);
    }
}