[BOJ/Python] 백준 9017번 - 크로스 컨트리

백준 #9017 크로스 컨트리

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

 

import sys
from collections import Counter
input = sys.stdin.readline

T = int(input())
for _ in range(T):
  N = int(input())
  teams = list(map(int, input().split()))
  counter = Counter(teams)
  scores = {}

  rank = 1
  for i in range(N):
    if counter[teams[i]] == 6:
      if teams[i] in scores:
        scores[teams[i]].append(rank)
      else:
        scores[teams[i]] = [rank]
      rank += 1
  
  print(sorted(scores, key = lambda x:(sum(scores[x][0:4]), scores[x][4]))[0])

 

 

 

---

 

•  Counter
  • 배열을 인자로 넘기면 각 원소가 몇 번씩 나오는지 객체(딕셔너리 형태) 반환

파이썬 collections 모듈의 Counter 사용하기