[백준] 8978 올림픽 (파이썬/python)

🎈문제

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

 

🎁알고리즘 및 접근

구현, 정렬

1. 금,은,동 메달 개수 순서로 정렬을 시켜놓은 뒤

2. 공동순위를 고려하여 순위를 매겨주면 되는 문제이다.

 

 

💻코드

# 8979 올림픽


# 입력 받기
n,k = map(int,input().split())
n_lst = [list(map(int,input().split())) for _ in range(n)]
# 금,은,동 메달 개수 순서로 정렬
n_lst = sorted(n_lst,key=lambda x:(x[1],x[2],x[3]),reverse=True)
# print(n_lst)

rank = 1    # 순위
cnt = 0     # 같은 순위 카운트
pre = n_lst[0][1:]  # 다음 순위와의 비교를 위해 메달 개수 저장

if n_lst[0][0] == k: # 1등이면 
    print(rank)
else:
    for i in range(1,n):
        if n_lst[i][1:] == pre: # 메달 개수가 같으면 공동순위 이므로
            cnt += 1    # 같은 순위 +1

        else:   # 메달 개수가 다르면 rank +
            pre = n_lst[i][1:]
            rank += cnt+1
            cnt = 0
        
        if n_lst[i][0] == k:
            print(rank)
            break

🎄코드 설명

 

이런 문제의 경우 마치 수학에서 점화식과 같이 내가 세운 규칙이 어디서부터 유효한지를 생각해보고 코드를 짜는 것이 중요하다. 

1. 정렬이 된 n_lst에서, 첫 번째 idx의 경우 두 번째 idx와 공동 순위일지라도 무조건 1등이다. 따라서 공동 순위를 따져볼 필요가 없다.

2. 이 문제의 핵심인 공동순위의 경우 두 번째 idx부터 적용된다. 

3. 따라서 1번째 idx의 정보를 미리 기억해놓고 2번째 idx 부터는 공동순위 여부를 따져보며 순위를 적용해준다.