[백준 2108] 통계학 with Python

문제 링크

www.acmicpc.net/problem/2108

2108번: 통계학

풀이

import sys

n = int(sys.stdin.readline())
lst = [0 for x in range(8001)]
for i in range(n):
    lst[int(sys.stdin.readline())+4000] += 1

# 산술평균
print(round(sum([(x-4000)*lst[x] for x in range(8001)])/n))

# 중앙값 
a, b, mid = 0, 0, n//2+1
if lst[0] >= mid:
    print(-4000)
else:
    for i in range(8000):
        a += lst[i]
        b = a + lst[i+1]
        if a < mid <= b:
            print(i+1-4000)
            break

# 최빈값
maximum = max(lst)
if lst.count(maximum) > 1:
    first_max = lst.index(maximum)
    lst[first_max] = 0
    print(lst.index(maximum)-4000)
    lst[first_max] = maximum
else:
    print(lst.index(maximum)-4000)

# 범위
a, b = 0, 8000
while lst[a] == 0: a += 1
while lst[b] == 0: b -= 1
print(b-a)

입력받는 수들은 -4000~4000까지의 정수이다. 리스트를 만들어서 각각의 정수가 몇 개 포함되어 있는지 센다.

 

산술평균은 정수를 정수의 개수만큼 곱한 것들의 합을 n으로 나눈 값을 출력하였다.

 

중앙값은 가장 작은 수인 -4000의 개수가 중앙값을 포함하면 바로 -4000을 반환하고, 그 외의 경우에는 그 다음으로 큰 수의 개수를 더했을 때 그 수가 중앙값을 포함하면 다음으로 큰 수를 중앙값으로 출력하였다.

 

최빈값이 1개일 때는 바로 출력, 2개 이상일 때에는 가장 작은 최빈값의 빈도수를 0으로 만들어놓고 다시 최빈값을 찾으면 두 번째로 작은 최빈값을 찾을 수 있다. 찾아서 반환한 후에는 아직 범위값을 구하지 않았으므로 가장 작은 최빈값의 빈도수를 원상복구한다.

 

범위는 가장 작은 수와 가장 큰 수부터 빈도수가 0이 아닌 첫 번째 수를 찾아 두 수의 차를 출력하였다.