[swea]1204. 최빈수 구하기, 파이썬(python)

문제 링크

 

학생들의 점수의 최빈값을 구하는 문제이다

 

• Idea : 계수정렬, 최빈값이 여러개일 때 인덱스 찾기

학생 점수의 최대값이 100이라고 계수정렬할 리스트를 [0]*101로 설정하지 않아도 된다, 받은 리스트 중 제일 큰 값에 1 만 더한 길이만큼으로 생성해도 충분하기 때문!

 

파이썬의 max와 index를 이용하면 제일 큰 값 중 제일 작은 인덱스를 알려주기에 인덱스가 여러개일 경우 아래와 같은 방법을 사용할 수 있다.

T = int(input())
for i in range(1, T+1):
    n = int(input())
    arr = list(map(int, input().split()))
    cnt_arr = [0] * (max(arr) + 1)
    for j in arr:
        cnt_arr[j] += 1

    tmp = [idx for idx, value in enumerate(cnt_arr) if max(cnt_arr) == cnt_arr[idx]]
    print(f"#{i} {tmp[-1]}")

끗