작심 24/7

[백준] 2108번 통계학 본문

백준

[백준] 2108번 통계학

모닝수박 2020. 5. 22. 18:08
 

2108번: 통계학

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

www.acmicpc.net

1. 산술평균은 배열을 입력 받을 때 sum에 계산해주고 N으로 나눈 값을 precision(0)을 이용해 출력한다.

 

2. 중앙값은 오름차순 정렬된 상태의 배열의 N/2번째 값을 출력한다.

 

3. 최빈값을 구하려면 먼저 배열을 오름차순으로 정렬해준 뒤 그 값에 따른 카운트를 저장해주기 위해 pair를 이용한다.

    vector를 선언하여 pair로 해당 수와 그 수의 카운트 값을 저장해준다.

    동시에 카운트 값의 최대 값을 cnt_max에 계속 갱신해준다.

    최빈값이 여러개 있을 경우엔 두번째로 작은 수를 출력해줘야 하기 때문에 cnt_max와 비교하여 찾아낸다.

 

4. 범위는 배열을 입력 받을 때 Max와 Min을 이용해 구해준 뒤 차이를 출력한다.

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
int arr[500001] = { 0 };
double sum = 0;
vector <pair<int,int>> v;

int main() {
	int N;
	cin >> N;
	int Max = -5000, Min = 5000;
	for (int n = 0; n < N; n++) {
		cin >> arr[n];
		sum += arr[n];
		Max = max(Max, arr[n]);
		Min = min(Min, arr[n]);
	}

	sort(arr, arr + N);

	int cnt = 1, cnt_max = 0;
	if (N == 1) {
		v.push_back(pair<int, int>(arr[0], cnt));
		cnt_max = cnt;
	}
	else {
		for (int i = 1; i < N; i++) {
			if (arr[i - 1] != arr[i]) {
				v.push_back(pair<int, int>(arr[i - 1], cnt));
				cnt_max = max(cnt_max, cnt);
				cnt = 0;
			}
			cnt++;
		}
			v.push_back(pair<int, int>(arr[N - 1], cnt));
			cnt_max = max(cnt_max, cnt);
	}

	int thisone = -1;
	for (int i = 0; i < v.size(); i++) {
		if (v[i].second == cnt_max) {
			if (thisone == -1) thisone = i;
			else {
				thisone = i;
				break;
			}
		}
	}
	cout << fixed;
	cout.precision(0);
	cout << sum / (double)N << "\n" << arr[N / 2] << "\n" << v[thisone].first << "\n" << Max - Min;

	return 0;
}
Comments