Submission #524934

#	Time	Username	Problem	Language	Result	Execution time	Memory
524934		Ziel	Diversity (CEOI21_diversity)	C++17	0 / 100	1 ms	204 KiB

diversity

#include <bits/stdc++.h>
using namespace std;
 
using ll = long long;
#define sz(x) (int)x.size()
 
signed main() {
	ios_base::sync_with_stdio(false);
	cin.tie(nullptr);
 
	int n, q;
	cin >> n >> q;
	vector<int> a(n + 1);
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
	}
	sort(a.begin() + 1, a.end());
	int _, __;
	cin >> _ >> __;

	set<int> s{a.begin() + 1, a.end()};
	int x = n - sz(s) + 1;

	ll ans = n * (n + 1) / 2ll;
	int c = 0, y = n - 1;
	for (int i = x + 1; i <= n; i++) {
		if (c == 2) {
			c = 0;
			y++;
		}
		ans += y;
		c++;
	}

	cout << ans;
	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