P1787 [入门赛 #22] 非众数 Hard Version.cpp

#include<bits/stdc++.h>
#define int int_fast64_t
#define lowbit(x) (x&(-x))
using namespace std;
int n,s[int(2e5+5)],p[int(1e5+5)];
void update(int x,int k){
    for(int i=x;i<(n<<1)+5;i+=lowbit(i)){
        s[i]+=k;
    }
}
int query(int x){
    int ans=0;
    for(int i=x;i>=1;i-=lowbit(i)){
        ans+=s[i];
    }
    return ans;
}
int32_t main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr),cout.tie(nullptr);
    string str;
    cin>>str;
    n=str.size();
    int ans=((n*(n+1))>>1);
    str=" "+str;
    for(char c='a';c<='z';c++){
        memset(s,0,sizeof s);
        int res=0;
        for(int i=1;i<=n;i++){
            p[i]=p[i-1]+(str[i]==c);
            update(i-(p[i-1]<<1)+n+1,1);
            res+=query((n<<1)+4)-query(i-(p[i]<<1)+n+2);
        }
        ans-=res;
    }
    cout<<ans;
    return 0;
}