Submission #593836

#	Time	Username	Problem	Language	Result	Execution time	Memory
593836		vladislav11	Diversity (CEOI21_diversity)	C++14	0 / 100	4 ms	316 KiB

diversity

#include <bits/stdc++.h>

using namespace std;

using ll = long long;

int n, q, l, r;
vector<int> a;

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

    cin >> n >> q;

    a.resize(n);
    for ( auto& el : a )
        cin >> el;

    cin >> l >> r;

    sort( a.begin(), a.end() );

    vector< pair<int,int> > b;
    b.push_back({ 1, a[0] });

    for ( int i=1; i<n; i++ )
    {
        if ( a[i] == b.back().second )
            b.back().first++;
        else
            b.push_back({ 1, a[i] });
    }

    sort( b.begin(), b.end(), greater< pair<int,int> >() );

    vector<int> c;

    while ( b.size() )
    {
        for ( int i=int(b.size())-1; i>=0; i-- )
        {
            c.push_back( b[i].second );
            b[i].first--;
        }

        while ( b.size() && b.back().first == 0 )
            b.pop_back();
    }

    /*for ( auto& el : c )
        cout << el << ' ';
    cout << '\n';*/

    ll ans = 0;

    for ( int i=0; i<n; i++ )
    for ( int j=i; j<n; j++ )
        ans += set<int>( c.begin()+i, c.begin()+j+1 ).size();

    cout << ans;

    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