메모리: 56548 KB, 시간: 492 ms
수학, 구현, 정렬
수를 처리하는 것은 통계학에서 상당히 중요한 일이다. 통계학에서 N개의 수를 대표하는 기본 통계값에는 다음과 같은 것들이 있다. 단, N은 홀수라고 가정하자.
N개의 수가 주어졌을 때, 네 가지 기본 통계값을 구하는 프로그램을 작성하시오.
첫째 줄에 수의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 단, N은 홀수이다. 그 다음 N개의 줄에는 정수들이 주어진다. 입력되는 정수의 절댓값은 4,000을 넘지 않는다.
첫째 줄에는 산술평균을 출력한다. 소수점 이하 첫째 자리에서 반올림한 값을 출력한다.
둘째 줄에는 중앙값을 출력한다.
셋째 줄에는 최빈값을 출력한다. 여러 개 있을 때에는 최빈값 중 두 번째로 작은 값을 출력한다.
넷째 줄에는 범위를 출력한다.
counter 라이브러리 사용법을 익히자
O(n)을 넘는 경우가 없기에 시간은 상관 안 하고 문제를 풀었다.
round(반올림 할 실수 n, 반올림할 소수 자리 k: 디폴트는 0)
import sys
from collections import Counter
n = int(sys.stdin.readline().rstrip())
data = []
for _ in range(n):
data.append(int(sys.stdin.readline().rstrip()))
# 산술 평균
print(round(sum(data) / n))
# 중앙값
data_sort = sorted(data) # 데이터 오름차순 정렬
print(data_sort[n // 2]) # 중간 인덱스의 값을 출력
# 최빈값
counter = Counter(data) # 데이터를 카운터로 생성
counter = list(counter.items()) # 생성된 카운터를 딕셔너리로 바꾸고 다시 리스트로 변환
counter.sort(key=lambda x: (-x[1], x[0])) # 개수 내림차순, 값 오름차순 정렬
if len(counter) >= 2 and counter[0][1] == counter[1][1]: # 최빈값이 여러개인 경우
print(counter[1][0])
else:
print(counter[0][0])
# 범위
print(max(data) - min(data))