제출 #572456

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
572456		valerikk	Diversity (CEOI21_diversity)	C++17	0 / 100	3 ms	1568 KiB

diversity

#include <bits/stdc++.h>

using namespace std;

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

	int n, Q;
	cin >> n >> Q;

	vector<int> a(n);
	for (int i = 0; i < n; ++i) {
		cin >> a[i];
	}

	vector<int> cnt(300001);
	for (int i = 0; i < n; ++i) {
		++cnt[a[i]];
	}

	vector<int> x;
	for (int i = 1; i <= 300000; ++i) {
		if (cnt[i] > 0) 
			x.push_back(i);
	}

	sort(x.rbegin(), x.rend());

	int k = x.size();

	deque<int> q;
	for (int i = 0; i < k; ++i) {
		if (i & 1)
			q.push_front(x[i]);
		else
			q.push_back(x[i]);
	}

	x.clear();
	for (int i : q)
		x.push_back(i);

	long long ans = n * 1ll * (n + 1) / 2 * k;
	
	long long sum = 0;
	for (int i = 0; i < k; ++i) {
		ans -= sum * (sum + 1) / 2;
		sum += x[i];
	}
	sum = 0;
	for (int i = k - 1; i >= 0; --i) {
		ans -= sum * (sum + 1) / 2;
		sum += x[i];
	}

	cout << ans << endl;


	return 0;
}

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