Submission #1114493

#	Time	Username	Problem	Language	Result	Execution time	Memory
1114493		vjudge1	Diversity (CEOI21_diversity)	C++17	64 / 100	179 ms	23536 KiB

diversity

#include <bits/stdc++.h>
#define pb push_back
#define spc << " " <<
#define endl "\n"
#define all(x) x.begin(), x.end()
#define int long long
#define ii pair<int, int>
#define vi vector<int>
#define vii vector<ii>
#define st first
#define nd second
#define mid (l+r)/2
#define inf 1e15
#define MOD 998244353
#define MX 200005
using namespace std;



void solve(){
    int n,q; cin >> n >> q;
    int arr[n+1];
    map<int, int> cnt;
    for(int i=1; i<=n; i++){
        cin >> arr[i];
        cnt[arr[i]]++;
    }
    for(int i=1; i<=q; i++){int a,b; cin >> a >> b;}
    vi wow;
    for(auto p:cnt) wow.pb(p.nd);
    sort(all(wow));
    int p1=0, p2=0, ans=n*(n+1)/2;
    for(int i=0; i<wow.size()-1; i++){
        if(p1>p2) swap(p1,p2);
        p1+=wow[i];
        ans+=p1*(n-p1);
    }
    cout << ans << endl;
}


signed main(){
    ios_base::sync_with_stdio(false);cin.tie(0);
    #ifdef Local
    freopen("in","r",stdin);
    freopen("out","w",stdout);
    /*#else
    freopen("248.in","r",stdin);
    freopen("248.out","w",stdout);*/
    #endif


    int t=1;
    //cin >> t;
    while(t--) solve();
}

Compilation message (stderr)

diversity.cpp: In function 'void solve()':
diversity.cpp:33:19: 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]
   33 |     for(int i=0; i<wow.size()-1; 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