[백준 / Python] 2108번 : 통계학

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

2108번: 통계학

 

---

---

첫번째 시도

import sys
input = sys.stdin.readline
 
n = int(input())
 
numList = [0] * n
 
for i in range(n):
    numList[i] = int(input())
 
numList.sort() 
 
numDict = {}
 
for k in numList:
    numDict[k] = numList.count(k)
 
maxIndex = [k for k,v in numDict.items() if max(numDict.values()) == v]
 
print(round(sum(numList)/n))
 
print(numList[(n//2)])
 
if len(maxIndex) >= 2:
    print(maxIndex[1])
else:
    print(maxIndex[-1])
 
print(max(numList)-min(numList))

 

시간초과....

최빈값을 찾기 위해 사용한 count 함수가 원인인 것 같다.

 

from collections import Counter
import sys
input = sys.stdin.readline
 
n = int(input())
 
numList = [0] * n
 
for i in range(n):
    numList[i] = int(input())
 
numList.sort()
 
counterList = Counter(numList).most_common(2)
 
print(round(sum(numList)/n))
 
print(numList[(n//2)])
 
try:
    if counterList[0][1] == counterList[1][1]:
        print(counterList[1][0])
    else:
        print(counterList[0][0])
except:
    print(counterList[0][0])
 
print(max(numList)-min(numList))

 

count를 사용하지 않고 최빈값을 찾기 위해 collections 모듈의 Counter 클래스를 사용했다. 그 중 내부 메소드인 most_common 함수를 사용하여 두 개의 최빈값과 빈도수를 새로 만든 변수에 저장하고 조건에 따라 출력을 다르게 해서 모든 조건에서 정답을 만족시키게끔 만들었다.

이번 문제에서 count를 사용하지 않아도 리스트의 각 요소 빈도를 계산할 수 있는 Counter 클래스에 대해 알게 되었다.