[Python] 백준 8979번 - 올림픽

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

8979번: 올림픽

막 어렵진 않은데 그냥 좀 복잡한 문제인 것 같다.

풀긴 했는데 주먹구구식으로 풀어서 코드가 하나도 안 예쁘다..

 

import sys
input = sys.stdin.readline

n, k = map(int, input().split())
medals = [[0, 0, 0, 0]] #국가, 금, 은, 동
rank = [0 for _ in range(n+1)]
cnt_rank = [0 for _ in range(n+1)] #같은 순위인 경우를 해결하기 위함
for _ in range(n):
    medals.append(list(map(int, input().split())))

medals.sort(key=lambda x:(x[1], x[2], x[3])) #금, 은, 동 순으로 정렬

for i in range(len(medals)-1, 0, -1):
    if i == len(medals) -1:
        rank[medals[i][0]] = 1
        cnt_rank[1] = 1
    elif medals[i][1] == medals[i+1][1]: #금메달 수 same
        if medals[i][2] == medals[i+1][2]: #은메달 수 same
            if medals[i][3] == medals[i+1][3]: #동메달 수 same
                rank[medals[i][0]] = rank[medals[i+1][0]]
                cnt_rank[rank[medals[i][0]]] += 1
            else:
            	#해당 국가보다 하나 높은 순위의 국가가 1개가 아닌 경우
                if cnt_rank[rank[medals[i+1][0]]] != 1:
                    #바로 전 순위의 개수(cnt_lank)를 고려하여 순위 계산 
                    rank[medals[i][0]] = rank[medals[i+1][0]] + cnt_rank[rank[medals[i+1][0]]]
                else:
                    rank[medals[i][0]] = rank[medals[i+1][0]] + 1
                cnt_rank[rank[medals[i][0]]] += 1
        else:
            if cnt_rank[rank[medals[i+1][0]]] != 1:
                rank[medals[i][0]] = rank[medals[i+1][0]] + cnt_rank[rank[medals[i+1][0]]]
            else:
                rank[medals[i][0]] = rank[medals[i+1][0]] + 1
            cnt_rank[rank[medals[i][0]]] += 1
    else:
        if cnt_rank[rank[medals[i+1][0]]] != 1:
            rank[medals[i][0]] = rank[medals[i+1][0]] + cnt_rank[rank[medals[i+1][0]]]
        else:
            rank[medals[i][0]] = rank[medals[i+1][0]] + 1
        cnt_rank[rank[medals[i][0]]] += 1

print(rank[k])