백준BOJ/Python : 2108번 통계학

백준BOJ/Python : 2108번 통계학

0

0

2108번 : 통계학 원본

알고리즘 분류

  • 수학
  • 구현
  • 정렬

문제

  1. 산술평균 : N개의 수들의 합을 N으로 나눈 값
  2. 중앙값 : N개의 수들을 증가하는 순서로 나열했을 경우 그 중앙에 위치하는 값
  3. 최빈값 : N개의 수들 중 가장 많이 나타나는 값
  4. 범위 : N개의 수들 중 최댓값과 최솟값의 차이 N은 홀수라고 가정하고, N개의 수가 주어졌을 때, 네 가지 기본 통계값을 구하는 프로그램을 작성하시오.

입력

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

셋째 줄에는 M(1 ≤ M ≤ 500,000)이 주어진다. 넷째 줄에는 상근이가 몇 개 가지고 있는 숫자 카드인지 구해야 할 M개의 정수가 주어지며, 이 수는 공백으로 구분되어져 있다. 이 수도 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다.

출력

첫째 줄에는 산술평균을 출력한다. 소수점 이하 첫째 자리에서 반올림한 값을 출력한다. 둘째 줄에는 중앙값을 출력한다. 셋째 줄에는 최빈값을 출력한다. 여러 개 있을 때에는 최빈값 중 두 번째로 작은 값을 출력한다. 넷째 줄에는 범위를 출력한다.

풀이

최빈값의 경우 값이 가장 많은 키들을 찾아 두 개 이상일 경우 두 번째로 작은 값을 출력하도록 하였다.

소스 코드

Python
from sys import stdin
input = stdin.readline

N = int(input())

mode = {}
array = []

# 입력한 숫자들 배열 그리고 딕셔너리 키로 저장
for i in range(N):
    num = int(input())
    array.append(num)
    if num in mode:
        mode[num] += 1
    else:
        mode[num] = 1
array = sorted(array)

# 산술평균 값 출력
print(round(sum(array)/N))

# 중앙값 출력
print(array[len(array) // 2])

# 최빈값 출력
modeNum = 0
for k in mode.keys():
    if mode[k] > modeNum:
        modeNum = mode[k]

temp = list(filter(lambda x: mode[x] == modeNum, mode.keys()))
temp = sorted(temp)

if len(temp) > 1:
    print(temp[1])
else:
    print(temp[0])

# 범위
print(array[-1] - array[0])

이 글이 도움이 되었나요?

신고하기
0분 전
작성된 댓글이 없습니다. 첫 댓글을 달아보세요!
    댓글을 작성하려면 로그인이 필요합니다.