[백준] 2108번: 통계학 python

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

2108번: 통계학

 

정답코드:

from collections import Counter
import sys

a = int(sys.stdin.readline())
b= []
for i in range(a):
  b.append(int(sys.stdin.readline()))

b.sort()

c = a//2 

# 산술평균
print(round(sum(b)/a))
# 중앙값
print(b[c])

# 최빈값
cnt = Counter(b).most_common(2)

if len(b)>1 and cnt[0][1] == cnt[1][1]:
    print(cnt[1][0])
else:
  print(cnt[0][0])

# 최댓값 - 최소값
print(b[a-1]-b[0])

 

나머지는 그럭저럭 할 만 했고

 

최빈도가 가장 문제였는데 count()함수를 사용하면 시간초과가 떠서 다른 방법을 생각해야 했다

검색해보니 collection의 Counter를 활용하여 빈도수를 구하는 것을 확인할 수 있었다

 

Counter(b)를 하면 dict 형태로 출력되기 때문에 .most_common()함수를 리스트에 담긴 튜플 형태로 요소들의 빈도를 구했다

 

5개의 수 1,3,8,-1,2를 입력한 경우 counter()와 counter().most_common()은 아래와 같다

print("Counter(b) = " ,Counter(b))
print("Counter(b).most_common(2) =" , Counter(b).most_common())

 

most_common()을 통해 만들어진 튜플은 (요소, 빈도수) 형태로 빈도수가 많은 수부터 작은수 순서대로 저장되어 있고, 같은 빈도수라면 처음 나온 값이 앞에 위치한다.

 

코드를 작성할 때 b.sort()로 정렬해줬기 때문에

최빈값 중 최소값: count[0][0]

최빈값 중 두번째로 작은 값: count[1][0]

임을 확인할 수 있다

 

참고한 블로그

https://joylee-developer.tistory.com/94

(파이썬) 백준 알고리즘 - 2108번 / Counter