0703_백준 알고리즘 #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
from collections import Counter

N = int(input())
num = [int(sys.stdin.readline()) for _ in range(N)]
num.sort()
mode = Counter(num).most_common()

print(round(sum(num) / N))
print(num[(N - 1) // 2])
if len(mode) > 1 and mode[0][1] == mode[1][1]:
    print(mode[1][0])
else:
    print(mode[0][0])
print(max(num) - min(num))

 

※ 최빈값 구하는 부분에서 Counter를 사용할 생각은 했지만 ! most_common()이라는 함수는 처음 사용해봤다. 

 

- most_common()

  -> 최빈값을 구하는 함수로 () 안에 숫자를 지정할 경우, 원하는 개수만큼 최빈값 출력 가능

num.sort()
mode = Counter(num).most_common()
if len(mode) > 1 and mode[0][1] == mode[1][1]:
    print(mode[1][0])
else:
    print(mode[0][0])

 

  -> 정렬을 하고 시작했기 때문에 count가 높은 순으로, 작은 수부터 출력됨

  -> (if) 두번째로 작은 값을 구하라고 했으므로 0, 1 인덱스의 count 횟수가 같은지 비교하여

      가장 높은 count가 2개 있을 경우,  0번과 1번 인덱스 중 1번 인덱스 출력

  -> (else) 가장 높은 count가 1개일 경우, 0번 인덱스 출력

 

++ 처음에 mode[0][1] == mode[1][1] 코드를 작성하지 않아 틀렸었다. most_common()이라는 새로운 함수를 학습할 수 있었던 문제 !