Python3/백준 알고리즘

2022.02.21 [백준] (python 파이썬) 통계학

ian's coding 2022. 2. 21. 23:20
728x90
반응형

 

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

 

 

 

 

728x90
반응형