Submission #881887

#	Time	Username	Problem	Language	Result	Execution time	Memory
881887		TAhmed33	Diversity (CEOI21_diversity)	C++98	64 / 100	102 ms	6796 KiB

diversity

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN = 3e5 + 25;
int n, a[MAXN], q, n1;
int freq[MAXN];
int main () {
	cin >> n >> q;
	for (int i = 1; i <= n; i++) cin >> a[i];
	sort(a + 1, a + n + 1);
	for (auto i : a) freq[i]++;
	vector <int> x;	
	for (int i = 1; i < MAXN; i++) {
		if (freq[i]) {
			x.push_back(freq[i]);
		}
	}
	int n1 = x.size();
	sort(x.begin(), x.end());
	deque <int> y;
	for (int i = n1 - 1; i >= 0; i -= 2) y.push_front(x[i]);
	for (int i = n1 - 2; i >= 0; i -= 2) y.push_back(x[i]);
	ll mn = 1e18;
	ll r = 0, sum = 0;
	for (auto i : y) {
		sum += r * (r + 1) / 2 + (n - i - r) * (n - i - r + 1) / 2;
		r += i;
	}
	sum = ll(n) * ll((n + 1)) / 2 * ((ll)y.size()) - sum;
	mn = sum;
	cout << mn << '\n';
}

• Subtask 1: 4 / 4
• Subtask 2: 10 / 10
• Subtask 3: 8 / 8
• Subtask 4: 16 / 16
• Subtask 5: 26 / 26
• Subtask 6: 0 / 36