[백준 2108] 통계학 (C++)

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

2108번: 통계학

문제

---

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

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

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

 

입력

---

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

 

출력

---

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

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

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

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

 

문제 풀이

---

문제에서 주어진 대로 구현하면 되는 문제.

 

평균은 모두 더해 N으로 나누어주면 되고, 중앙값은 정렬한 배열의 N/2번째 인덱스의 값을 출력하면 된다.

범위는 정렬한 배열의 맨 끝 값에서 맨 첫번째 값를 빼주면 된다.

최빈값의 경우에는 map을 이용했는데, map에 횟수를 저장한 후 map 안에 들어있는 모든 원소들을 탐색하며 가장 많이 나오는 수들을 vector에 저장하였다. 배열의 탐색이 끝나면 vector를 정렬하고, 벡터에 들어있는 원소 중 두 번째로 작은 값을 출력하도록 하였다.

 

평균의 경우에는 float를 사용하면 부동 소수점 오류 때문에 값이 달라질 수 있으므로 double을 사용해야 한다. 또한 반올림 함수인 round를 사용할 경우 -0이 나오는 것도 가능하므로 이를 다른 변수에 저장하여 출력해주어야 한다.

 

아래는 코드.

#include <iostream>
#include <algorithm>
#include <cmath>
#include <vector>
#include <map>
using namespace std;

int main()
{
	cin.tie(NULL);
	ios::sync_with_stdio(false);

	int N;
	cin >> N;
	int* arr = new int[N];
	map<int, int> m;
	map<int, int>::iterator it;
	vector<int> maxCountV;
	double sum = 0;
	int maxCount = 0;
	for (int i = 0; i < N; i++)
	{
		cin >> arr[i];
		sum += arr[i];
		it = m.find(arr[i]);
		if (it == m.end())
		{
			m.insert({ arr[i], 1 });
		}
		else
		{
			it->second++;
		}
	}
	for (it = m.begin(); it != m.end(); it++)
	{
		if (it->second > maxCount)
		{
			maxCountV.clear();
			maxCountV.push_back(it->first);
			maxCount = it->second;
		}
		else if (it->second == maxCount)
		{
			maxCountV.push_back(it->first);
		}
	}
	sort(maxCountV.begin(), maxCountV.end());
	sort(arr, arr + N);
	int average = round(sum / (double)N);
	int median = arr[N / 2];
	int mode = 0;
	int range = arr[N - 1] - arr[0];
	if (maxCountV.size() == 1)
	{
		mode = maxCountV.at(0);
	}
	else
	{
		mode = maxCountV.at(1);
	}

	cout << average << "\n";
	cout << median << "\n";
	cout << mode << "\n";
	cout << range << "\n";
	return 0;
}