[백준]8979번: 올림픽

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

8979번: 올림픽

---

문제에서 조심해야 할 조건

 -공동순위가 있을 경우 1등, 2등, 2등, 3등 이런식으로 순위가 매겨지는 게 아님. 1등, 2등, 2등, 4등 으로 매겨진다.

 -국가번호가 1부터 순서대로 입력되지 않을 수 있다.

 

개인적으로 위 조건들을 잘못 읽어서 정답판정 받는 데에 생각보다 시간이 좀 걸렸다.

문제를 똑바로 읽쟈..

---

정답 코드

import sys
input=sys.stdin.readline

n,k = map(int,input().split())
medals=[]
for i in range(n):
  medals.append(list(map(int,input().split())))

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

medals[0].append(1)
rank=1
same=1
for i in range(1,len(medals)):
  if medals[i][1]==medals[i-1][1] and medals[i][2]==medals[i-1][2] and medals[i][3]==medals[i-1][3]:
    same += 1
    medals[i].append(rank)
  else:
    rank += same #동순위만큼 순위 증가시켜주기
    same = 1
    medals[i].append(rank)

for i in medals:
  if i[0]==k:
    print(i[4])
    exit()