Submission #1103434

#	Time	Username	Problem	Language	Result	Execution time	Memory
1103434		ro9669	Diversity (CEOI21_diversity)	C++17	64 / 100	7044 ms	6008 KiB

diversity

#include <bits/stdc++.h>
#define fi first
#define se second
#define all(v) v.begin() , v.end()
#define sz(v) int(v.size())
#define unq(v) sort(all(v)); v.resize(unique(all(v)) - v.begin());
using namespace std;

typedef long long ll;
typedef pair<int , int> ii;
typedef pair<long long , int> lli;

const int maxN = int(3e5)+7;

int n , q , a[maxN] , cnt[maxN] , sum[2];

void solve(){
    cin >> n >> q;
    for (int i = 1 ; i <= n ; i++) cin >> a[i];
    while (q--){
        int l , r;
        cin >> l >> r;
        for (int i = 1 ; i < maxN ; i++) cnt[i] = 0;
        for (int i = l ; i <= r ; i++) cnt[a[i]]++;
        vector<int> p;
        for (int i = 1 ; i < maxN ; i++){
            if (cnt[i] > 0) p.push_back(cnt[i]);
        }
        sort(all(p));
        ll ans = 0;
        sum[0] = sum[1] = 0;
        for (int i = 0 ; i < sz(p) ; i++){
            ans += 1ll * p[i] * (p[i] + 1) / 2;
            if (i % 2 == 0){
                int x = sum[0] , y = (r - l + 1) - sum[0] - p[i];
                ans += 1ll * x * y;
                ans += 1ll * (x + y) * p[i];
            }
            else{
                int x = sum[1] , y = (r - l + 1) - sum[1] - p[i];
                ans += 1ll * x * y;
                ans += 1ll * (x + y) * p[i];
            }
            sum[i % 2] += p[i];
        }
        cout << ans << "\n";
    }
}

#define name "A"

int main(){
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    if (fopen(name".INP" , "r")){
        freopen(name".INP" , "r" , stdin);
        freopen(name".OUT" , "w" , stdout);
    }
    int t = 1; //cin >> t;
    while (t--) solve();
    return 0;
}

Compilation message (stderr)

diversity.cpp: In function 'int main()':
diversity.cpp:55:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   55 |         freopen(name".INP" , "r" , stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~
diversity.cpp:56:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   56 |         freopen(name".OUT" , "w" , stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~

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