알고리즘 분류
- 수학
- 구현
- 정렬
문제
- 산술평균 : N개의 수들의 합을 N으로 나눈 값
- 중앙값 : N개의 수들을 증가하는 순서로 나열했을 경우 그 중앙에 위치하는 값
- 최빈값 : N개의 수들 중 가장 많이 나타나는 값
- 범위 : 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])
Ghost