백준/2108번 :: 통계학 (Python, 파이썬, 알고리즘)

 

문제

---

수를 처리하는 것은 통계학에서 상당히 중요한 일이다. 통계학에서 N개의 수를 대표하는 기본 통계값에는 다음과 같은 것들이 있다. 단, N은 홀수라고 가정하자.

1. 산술평균 : N개의 수들의 합을 N으로 나눈 값
2. 중앙값 : N개의 수들을 증가하는 순서로 나열했을 경우 그 중앙에 위치하는 값
3. 최빈값 : N개의 수들 중 가장 많이 나타나는 값
4. 범위 : N개의 수들 중 최대값과 최소값의 차이

N개의 수가 주어졌을 때, 네 가지 기본 통계값을 구하는 프로그램을 작성하시오.

 

 

입력

---

첫째 줄에 수의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 그 다음 N개의 줄에는 정수들이 주어진다. 입력되는 정수의 절대값은 4,000을 넘지 않는다.

 

 

출력

---

첫째 줄에는 산술평균을 출력한다. 소수점 이하 첫째 자리에서 반올림한 값을 출력한다.

둘째 줄에는 중앙값을 출력한다.

셋째 줄에는 최빈값을 출력한다. 여러 개 있을 때에는 최빈값 중 두 번째로 작은 값을 출력한다.

넷째 줄에는 범위를 출력한다.

 

 

예제

---

예제입력1

5
1
3
8
-2
2

 

예제출력1

2
2
1
10

 

예제입력2

1
4000

 

예제출력2

4000
4000
4000
0

 

 

예제입력3

5
-1
-2
-3
-1
-2

 

예제출력3

-2
-2
-1
2

 

 

파이썬 코드

---

# -*- coding: utf-8 -*-
# Python 3.4.5
import sys
from collections import Counter

n = int(sys.stdin.readline())
data = []
my_sum = 0
for i in range(n) :
    temp = int(sys.stdin.readline())
    data.append(temp)
    my_sum += temp # 합
print('%.0f' % (my_sum/n))

# n이 1일때는 굳이 연산할 필요가 없음
if n == 1:
    print(data[0])
    print(data[0])
    print(0)
    sys.exit()

# 리스트 정렬
data = sorted(data)
print(data[n//2])

# 사전 정렬
cnt = Counter(data)
cnt = sorted(cnt.items(), key=lambda x: (-x[1],x[0]))

# 최빈값 출력
if cnt[0][1] == cnt[1][1] :
    print(cnt[1][0])
else :
    print(cnt[0][0])

# 범위
print(data[-1]-data[0])​

 

최빈값 빼고는 간단했던 문제인것 같다.

 

최빈값을 구할때, list형태로 값을 저장을 했다.(코드 7번째 줄, 11번째 줄) 그리고 collections 모듈의 Counter() 메소드를 이용해서 data안의 원소들의 개수를 딕셔너리 형태로 저장했다. 예를들어, {'1':2, '3':'1'....}와 같은 형태로 저장. (코드 27번째 줄)

 

마지막으로 sorted() 메소드를 이용해서 사전을 정렬했다. 우선순위로 갯수는 내림차순으로, 원소는 오름차순으로 정렬했다. (코드 28번째줄)

 

결과

---

 

 

 

※ 직접 문제 풀고 돌려본 뒤, 채점까지 마친 후에 작성한 글입니다.

더 좋은 방법이 있다면, 댓글로 알려주시면 감사하겠습니다 :)

 

문제 출처

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