STUDY/Algorithm

[백준] 8979 올림픽

sinawi95 2021. 2. 23. 22:17
728x90

https://acmicpc.net/problem/8979 

 

8979번: 올림픽

입력의 첫 줄은 국가의 수 N(1 ≤ N ≤ 1,000)과 등수를 알고 싶은 국가 K(1 ≤ K ≤ N)가 빈칸을 사이에 두고 주어진다. 각 국가는 1부터 N 사이의 정수로 표현된다. 이후 N개의 각 줄에는 차례대로 각

www.acmicpc.net

import sys
input = sys.stdin.readline

n, k = map(int, input().split())
medal = list()
k_medal = tuple()
for _ in range(n):
    tmp = tuple(map(int, input().split()))
    if tmp[0] == k:
        k_medal = tmp
    else:
        medal.append(tmp)
print(medal,k_medal)
up = []
for md in medal:
    if md[1] == k_medal[1]:
        if md[2] == k_medal[2]:
            if md[3] > k_medal[3]:
                up.append(md)
        elif md[2] > k_medal[2]:
            up.append(md)
    elif md[1] > k_medal[1]:
        up.append(md)
print(len(up)+1)

쏘 이즤~~~~~~

입력을 튜플로 받고 k보다 높은 등수의 개수에 +1 해주면 된다.