백준 2108 - 통계학

문제 번호: 2108

문제 제목: 통계학

문제 주소: https://www.acmicpc.net/problem/2108

문제 내용

입력된 N개의 수에 대한 산술평균, 중앙값, 최빈값, 범위를 출력한다.
산술평균 - 소수 첫째 자리에서 반올림
중앙값 - 요소가 홀수 일 때는 중앙에 위치한 값, 짝수일 때는 중앙에 위치한 두 수의 평균
최빈값 - 가장 많이 출현한 수. 여러 개 있을 경우 두 번째로 작은 값

테스트 케이스

5 1 3 8 -2 2	2 2 1 10
1 4000	4000 4000 4000 0
5 -1 -2 -3 -1 -2	-2 -2 -1 2

문제 풀이

입력하는 횟수가 50만개로 많고, 반대로 정수의 범위는 -4000 ~ 4000으로 좁은 편이다. 
입력 순서를 고려하지 않아도 되고 중복값이 존재하므로 인덱스를 만들고 이를 바탕으로 계산하는 것이 좋다.

산술평균 - 합계가 양수일 경우에는 +0.5를, 음수일 경우에는 -0.5를 한 후에 int로 형변환하면 된다.
중앙값 - N / 2개에 해당하는 인덱스 값을 출력한다.
최빈값 - 오름차순으로 인덱스를 탐색한다. 현재 최빈값과 같을 경우 중복 여부를 확인하고, 더 높은 수가 나오면 중복 체크를 초기화한다.
최빈값 - 입력받는 시점에 min, max를 구하고 max - min을 출력한다.

풀이 코드

#include <iostream>
using namespace std;

void BAEKJOON_2108()
{
	int N;
	cin >> N;
	int32_t index[8001] = { 0 };
	int32_t sum = 0;
	int16_t min = 4000;
	int16_t max = -4000;
	for (int i = 0; i < N; ++i)
	{
		int16_t input;
		cin >> input;
		if (min > input)
		{
			min = input;
		}
		if (max < input)
		{
			max = input;
		}
		sum += input;
		index[input + 4000]++;
	}
	// 산술평균
	double average = (double)sum / N;
	average = average > 0 ? average + 0.5 : average - 0.5;
	cout << static_cast<int32_t>(average * 10) / 10 << endl;
	// 중앙값
	int32_t middle = N / 2;
	for (int i = 0; i < 8001; i++)
	{
		middle -= index[i];
		if (middle < 0)
		{
			middle = i - 4000;
			break;
		}
	}
	cout << middle << endl;
	// 최빈값
	int16_t mode;
	int32_t count = 0;
	bool overlap = false;
	for (int i = 0; i < 8001; ++i)
	{
		if (count < index[i])
		{
			count = index[i];
			mode = i - 4000;
			overlap = false;
		}
		else if (count == index[i] && !overlap)
		{
			mode = i - 4000;
			overlap = true;
		}
	}
	cout << mode << endl;
	// 범위
	cout << max - min << endl;
	return;
}