2022.02.21 [백준] (python 파이썬) 통계학
https://www.acmicpc.net/problem/2108
2108번: 통계학
첫째 줄에 수의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 단, N은 홀수이다. 그 다음 N개의 줄에는 정수들이 주어진다. 입력되는 정수의 절댓값은 4,000을 넘지 않는다.
www.acmicpc.net
풀이
1. 산술평균구하기
sum함수를 이용하여 arr배열에 있는 값을 모두 더한 뒤, arr의 길이만큼 나눠준다. 이 값을 round함수를 이용하여
소수 첫번째 자리에서 반올림해준다.
2. 중앙값 구하기
문제에서 n은 홀수라고 정했기 때문에 배열을 오름차순 정렬해준 뒤, (arr배열의 길이//2)번째 인덱스가 중앙값이다.
3. 최빈값 구하기
최빈값은 파이썬에서 Counter함수를 이요하여 구할 수 있지만 Counter함수를 이용하지 않고 구해보았다.
각 숫자를 카운트할 cnt배열을 만들었다. 크기가 8001인 이유는 문제에서 정수의 범위는 절대값4000을 넘지않는다 고 했기 때문이다. 그리고 for문을 돌려 arr배열안에 있는 값을 cnt를 이용하여 카운트했다. 이때 i+4000을 한 이유는
인덱스가 음수가 될 수 없기 때문이다. 그리고 최빈값을 구한 뒤, for문을 다시 돌려 최빈값을 가진 인덱스 값을 모두 max_cnt에 담고 길이가 1이하면 0번째 인덱스 값을 출력 1보다 크면 최빈값이 여러개이기 때문에 두번째로 작은
1번째 인덱스 값을 출력함
4. 범위 구하기
arr[-1]을 하면 뒤에서 첫번째 인덱스를 불러 올 수 있기 때문에 arr[-1] - arr[1]을 했다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
import sys
input=sys.stdin.readline
n=int(input())
arr=[]
for i in range(n):
arr.append(int(input()))
arr.sort()
#산술평균
print(round(sum(i for i in arr)/len(arr)))
#중앙값
print(arr[len(arr)//2])
#최빈값
cnt=[0]*8001
max_cnt=[]
for i in arr:
cnt[i+4000]+=1
a=max(cnt)
for i in range(len(cnt)):
if cnt[i]==a:
max_cnt.append(i-4000)
max_cnt.sort()
if len(max_cnt)>1:
print(max_cnt[1])
else:
print(max_cnt[0])
#범위
print(arr[-1]-arr[0])
|
cs |