Submission #1229038

#	Time	Username	Problem	Language	Result	Execution time	Memory
1229038		raspy	Diversity (CEOI21_diversity)	C++20	0 / 100	7 ms	4936 KiB

diversity

#include <bits/stdc++.h>

#define vi vector<int>

#define int long long

using namespace std;

const int N = 3e5+5;

int ste[N];
int prs[N];

signed main()
{
	for (int i = 1; i < N; i++)
		prs[i]=prs[i-1]+i;
	int n, q;
	cin >> n >> q;
	vi a(n);
	for (int&v:a)
	{
		cin >> v;
		ste[v]++;
	}
	cin >> q >> q;
	sort(ste, ste+N);
	int stl = 1, rez=0;
	for (int i = 0; i < N; i++)
	{
		if (ste[i] == 0)
			continue;
		int std = n-stl+1;
		rez += stl*std;
		rez += prs[std-1]-prs[std-ste[i]];
		stl += ste[i];
	}
	cout << rez << "\n";
	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