Submission #1068516

#	Time	Username	Problem	Language	Result	Execution time	Memory
1068516		Andrey	Diversity (CEOI21_diversity)	C++14	64 / 100	37 ms	7068 KiB

diversity

#include<bits/stdc++.h>
using namespace std;

vector<long long> br(300001);

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    long long n,q;
    cin >> n >> q;
    long long ans = 0;
    vector<long long> haha(n+1);
    for(long long i = 1; i <= n; i++) {
        cin >> haha[i];
        br[haha[i]]++;
    }
    sort(br.begin(),br.end());
    long long l = 0,r = 0;
    for(long long i = 0; i < br.size(); i++) {
        if(br[i] > 0) {
            if(l > r) {
                swap(l,r);
            }
            long long c = n-l-br[i];
            ans+=n*(n+1)/2-l*(l+1)/2-c*(c+1)/2;
            l+=br[i];
        }
    }
    cout << ans;
    return 0;
}

Compilation message (stderr)

diversity.cpp: In function 'int main()':
diversity.cpp:21:28: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   21 |     for(long long i = 0; i < br.size(); i++) {
      |                          ~~^~~~~~~~~~~

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