Submission #659720

#	Time	Username	Problem	Language	Result	Execution time	Memory
659720		fatemetmhr	Diversity (CEOI21_diversity)	C++17	0 / 100	6 ms	1880 KiB

diversity

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

#define pb     push_back
#define fi     first
#define se     second
#define all(x) x.begin(), x.end()

const int maxn5 = 4e5 + 10;

int cnt[maxn5];

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

    int n, q; cin >> n >> q;
    for(int i = 0; i < n; i++){
        int a; cin >> a;
        cnt[a]++;
    }
    sort(cnt, cnt + maxn5);
    ll ans = 0;
    int a; cin >> a >> a;
    ll ind = 1;
    for(int i = 0; i < maxn5; i++) if(cnt[i]){
        ans += ind * (n + 1 - ind);
        ind++; cnt[i]--;
        while(cnt[i]--){
            ans += (n + 1 - ind);
            ind++;
        }
    }
    cout << ans << endl;

}

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