본문 바로가기
알고리즘/백준온라인저지 (Python)

[백준온라인저지] 8979 - 올림픽

by 미래문 2021. 8. 24.
반응형


 

 

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

 

8979번: 올림픽

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

www.acmicpc.net

import sys

n, k = map(int, sys.stdin.readline().split(' '))
medals = [ list(map(int, sys.stdin.readline().split(' '))) for _ in range(n)]

medals.sort(key=lambda x : (-x[1], -x[2], -x[3]))

def is_equal(medals, sum):
    for i in range(len(medals)):
        tmp_sum = medals[i][1] * 100 + medals[i][2] * 10 + medals[i][3]
        
        if tmp_sum == sum:
            return i
    return -1

for i in range(len(medals)):
    if medals[i][0] == k:
        sum = medals[i][1] * 100 + medals[i][2] * 10 + medals[i][3]

        print(is_equal(medals, sum) + 1)
반응형

댓글