백준 #8979 - [S4] 올림픽 : 정렬

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

8979번: 올림픽

 

(python)

파이썬에서 제공하는 sort 를 활용하여 간단하게 해결해볼 수 있습니다.

https://jamong-5.tistory.com/55

[sort] 파이썬에서 커스텀한 기준으로 이터러블한 객체 정렬하기

 

sort 후에 등수 계산을 잘 처리해주면 되는 문제입니다.

금은동 갯수가 똑같은 경우, 같은 등수로 처리하고 뒤에 등수를 비워주어야 하는 조건을 잘 처리해줍시다.

def solution(input) : 
	# 입력 받기
    N,K = list(map(int,input().split()))
    country = []
    for _ in range(N) :
        country.append(list(map(int, input().split())))
    # 커스텀 정렬
    country.sort(key=lambda x: (x[1]*-1, x[2]*-1, x[3]*-1))
	
    # 등수 계산
    i = 0
    s = 0
    for n in range(N) :
        i+=1
        if n == 0 :
            if country[n][0] == K :
                print(i)
                return
            continue
        if country[n][1:] == country[n-1][1:] :
            i-=1
            s+=1
        elif s > 0 :
            i+=s
            s=0
        if country[n][0] == K :
            print(i)
            return

solution(input)