Submission #1186745

#	Time	Username	Problem	Language	Result	Execution time	Memory
1186745		Unforgettablepl	Diversity (CEOI21_diversity)	C++20	14 / 100	7094 ms	2632 KiB

diversity

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

#define int long long

const int modulo = 1e9+7;


int32_t main(){
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    int N,Q;
    cin >> N >> Q;
    vector<int> arr(N);
    for(int&i:arr)cin>>i;
    vector<int> newarr;
    {
        int c=0;
        map<int,int> comp;
        for(int&i:arr){
            if(!comp.count(i)){
                comp[i]=c++;
                newarr.emplace_back(0);
            }
            newarr[comp[i]]++;
        }
    }
    sort(newarr.begin(),newarr.end());
    auto test = [&](){
        int ans = 0;
        int curr = 0;
        for(int&i:newarr){
            int L = curr+1;
            int R = curr+i;
            curr = R;
            ans+=R*(N+1ll) - L*L + ((L+1ll)*L - R*(R+1ll))/2ll;
        }
        return ans;
    };
    int minima = 1e17;
    do{
        minima=min(minima,test());
    } while(next_permutation(newarr.begin(),newarr.end()));
    cout << minima << '\n';
}

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