[백준 🥈4] 9017번 크로스 컨트리 (Python/파이썬)

1️⃣ 문제

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

2️⃣ 코드

참가자 수가 6명 보다 작은 팀은 반드시 점수 계산에서 제외해줘야 제대로 된 결과가 나온다
이 조건 무시해서 처음에 테케부터 틀렸음

t = int(input())
for _ in range(t):
    n = int(input())
    count = {}   # 팀 당 주자 수
    temp = list(map(int, input().split()))
    
    # 팀 별로 주자 수 세기
    for i in range(n):
        if temp[i] in count:
            count[temp[i]] += 1
        else:
            count[temp[i]] = 1
    
    # 제외되는 팀 구하기
    dele = {}   # 제외되는 팀 
    for k, v in count.items():
        if v < 6:
            dele[k] = 1
    
    # 점수 계산
    team = {}
    idx = 1    # 점수
    for i in range(n):
        if temp[i] not in dele:
            if temp[i] in team:
                if team[temp[i]][0] < 4:
                    team[temp[i]][0] += 1
                    team[temp[i]][1] += idx
                elif team[temp[i]][0] == 4:
                    team[temp[i]][0] += 1
                    team[temp[i]][2] = idx
            else:
                team[temp[i]] = [1, idx, 0]   # [팀당 주자 수, 상위 4명 점수 합, 5번째 주자의 점수]

            idx += 1
    
    # 순위 정렬
    team = sorted(team.items(), key=lambda x:(x[1][1], x[1][2]))

메모리 제한이 작아서 걱정했지만 범위가 작아서 그런지 무리없이 통과할 수 있었다
근데 실버4 수준은 아닌 것 같은뎅