Muutokset

#include <bits/stdc++.h>
using namespace std;
 
typedef long long ll;
 
ll tr[101010], tl[101010], a[101010];
ll cnt_right[101010];
ll cumul_left[101010], cumul_right[101010];
ll ans_left=0, ans_right=0;

int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
    int n, m;
    cin >> n >> m;
    
    for (int i=1; i<=m; ++i){
        cin >> a[i];
        tr[a[i]]++;
    }
    
    for (int i=1; i<=n; ++i){
        cnt_right[tr[i]]++;
    }
    for (int i=m; i>=1; --i){
        cumul_right[i] = cumul_right[i+1]+cnt_right[i];
        if (cnt_right[i]){
            ans_right+=i*(((cumul_right[i+1]+1)+cumul_right[i])*cnt_right[i])/2;
        }
    }
    for (int i=1; i<=m-1; ++i){
        ll b = tr[a[i]];
        ll c = tl[a[i]];
        tr[a[i]]--;
        ll d = cumul_right[b];
        ans_right -= d;
        cumul_right[b]--;
        
        tl[a[i]]++;
        cumul_left[c+1]++;
        ll e = cumul_left[c+1];
        ans_left += e;
        cout << ans_left + ans_right << " ";
    }   
    cout << "\n";
}
 

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

Test 4

Group: 1, 2, 3

Verdict: ACCEPTED

Test 5

Group: 1, 2, 3

Verdict: ACCEPTED

Test 6

Group: 2, 3

Verdict: ACCEPTED

Test 7

Group: 2, 3

Verdict: ACCEPTED

Test 8

Group: 2, 3

Verdict: ACCEPTED

Test 9

Group: 2, 3

Verdict: ACCEPTED

Test 10

Group: 3

Verdict: ACCEPTED

Test 11

Group: 3

Verdict: ACCEPTED

Test 12

Group: 3

Verdict: ACCEPTED

Test 13

Group: 3

Verdict: ACCEPTED