240703 [BOJ/백준] 2108. 통계학

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

---

문제

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

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

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

---

입력

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

---

출력

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

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

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

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

---

제출코드

import sys

n = int(input())
numbers = [int(sys.stdin.readline()) for _ in range(n)]

numbers.sort() # 정렬하면 중앙값 범위 구하기 쉬울 듯
dict = {}
mode_key = []
for number in numbers:  # 숫자마다 몇 개 있는지 딕셔너리로 저장
    if number not in dict:
        dict[number] = 1
    else:
        dict[number] += 1

print(round(sum(numbers)/len(numbers)))
print(numbers[len(numbers)//2])
mode = max(dict.values()) # 최빈값
for k, v in dict.items():
    if v == mode:
        mode_key.append(k)
if len(mode_key) > 1:
    print(mode_key[1]) # 두번째로 작은 수 출력
else:
    print(mode_key[0])
print(numbers[-1]-numbers[0])

결과

---

처음 시간초과는 입력값이 최대 50만개인데 readline을 안써서 맨 첫 문제도 채점하지 못하고 시간초과가 되었다. 바로 넣었으면 한 번에 맞추는건데...ㅜㅜ 출력초과는 readline 잘 적용됐나 보려고 print한걸 안지우는 실수를 했다...ㅜㅜ 아쉽네ㅜㅜ 아무튼 시간초과를 고려해 딕셔너리를 써서 문제해결!