[Python] 백준 2108번 - 통계학

문제 https://www.acmicpc.net/problem/2108

2108번: 통계학

 

딕셔너리 value값 기준으로 정렬할 때 lambda를 사용해야 하는지 몰랐다.

lambda 사용해서 정렬하면 튜플 형태로 바뀌기 때문에 유의해야 한다.

 

정렬 방법 모를 때는 웬만하면 lambda로 접근하면 된다고 생각해도 되려나.

https://hello-bryan.tistory.com/78

[Python] Dictionary Sort (정렬) lambda

 

최빈값 더 쉽게 구하는 방법이 있는지 찾아봐야겠다.

코드가 좀 복잡한 것 같다.

 

import sys
input = sys.stdin.readline

n = int(input())
num = []
for _ in range(n):
    num.append(int(input()))

num.sort()
print(round(sum(num)/n)) #산술평균
print(num[(n-1)//2]) #중앙값

#최빈값
cnt = {}
for i in range(n):
    if num[i] not in cnt.keys():
        cnt[num[i]] = 1
    else:
        cnt[num[i]] += 1
cnt = sorted(cnt.items(), key = lambda x:(-x[1], x[0]))
for i in range(len(cnt)):
    cnt[i] = list(cnt[i])
if len(cnt) == 1 or cnt[0][1] != cnt[1][1]:
    print(cnt[0][0])
else:
    print(cnt[1][0])

print(max(num) - min(num)) #범위