Submission #1318308

#	Time	Username	Problem	Language	Result	Execution time	Memory
1318308		spetr	Diversity (CEOI21_diversity)	C++20	0 / 100	0 ms	332 KiB

diversity

#include <bits/stdc++.h>

using namespace std;

#define ll long long
const ll mmod = 998244353;  
#define vl vector<long long>
#define vll vector<vector<long long>>
#define pl pair<long long, long long>
#define vb vector<bool>

int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    ll n, q;
    cin >> n >> q;
    vl nums (n);
    for (ll i = 0; i < n; i++){cin >> nums[i];}
    ll l, r;
    cin >> l >> r;

    vl dp (n, 1);
    sort(nums.begin(), nums.end());

    for (ll i = 1; i < n; i++){
        if (nums[i] == nums[i-1]){
            dp[i] = dp[i-1]+1;
        }
        else{
            dp[i] = dp[i-1] + 1 + i;
        }
    }
    ll suma = 0;
    for (ll i = 0; i < n; i++){suma += dp[i];}
    cout << suma << "\n";


    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