홍우진의 개발 일기장

[백준] 8979번: 올림픽/ 파이썬 본문

알고리즘/백준

[백준] 8979번: 올림픽/ 파이썬

홍우진 2022. 9. 8. 23:58
728x90
반응형

문제 링크


https://www.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())
s = []
for i in range(n):
    s.append(list(map(int, input().split())))
s.sort(key=lambda x : (-x[1], -x[2], -x[3]))
for i in range(n):
    if s[i][0] == k:
        index = i
for i in range(n):
    if s[index][1:] == s[i][1:]:
        print(i + 1)
        break

코드 해석


입력을 받은 뒤

금메달이 많은 순으로 정렬하고 수가 같다면 은메달, 수가 같다면 동메달이 많은 순으로 정렬하여준다.

 

728x90
반응형
Comments