后缀数组+主席树

https://vjudge.net/problem/Gym-102302K

后缀数组去重，主席树查询区间内在l,r之间的数的个数

#include<iostream>
#include<stack>
#include<algorithm>
#include<stdio.h>
#include<string.h>
#include<cmath>
#include<map>
using namespace std;
#define inf 0x3f3f3f3f
#define ll long long
#define maxn 500010
#define mod 10007
int n;
ll ch[maxn];//从0开始存储
int Rank[maxn],a[maxn],b[maxn],cnta[maxn],cntb[maxn],sa[maxn],tsa[maxn],hight[maxn];
//sa[i] 排名为i的起点（从1开始）
void solve()
{
//    for(int i=0; i<=256; i++)cnta[i]=0;
    memset(cnta,0,sizeof(cnta));
    for(int i=1; i<=n; i++)cnta[ch[i-1]]++;
    for(int i=1; i<=500005; i++)cnta[i]+=cnta[i-1];
    for(int i=n; i>0; i--)sa[cnta[ch[i-1]]--]=i;
    Rank[sa[1]]=1;
    for(int i=2; i<=n; i++)
    {
        Rank[sa[i]]=Rank[sa[i-1]];
        if(ch[sa[i]-1]!=ch[sa[i-1]-1])Rank[sa[i]]++;
    }
    for(int l=1; Rank[sa[n]]<n; l<<=1)
    {
        memset(cnta,0,sizeof(cnta));
        memset(cntb,0,sizeof(cntb));
        for(int i=1; i<=n; i++)
        {
            cnta[a[i]=Rank[i]]++;
            cntb[b[i]=(i+l<=n)?Rank[i+l]:0]++;
        }
        for(int i=1; i<=n; i++)
        {
            cnta[i]+=cnta[i-1];
            cntb[i]+=cntb[i-1];
        }
        for(int i=n; i>0; i--)tsa[cntb[b[i]]--]=i;
        for(int i=n; i>0; i--)sa[cnta[a[tsa[i]]]--]=tsa[i];
        Rank[sa[1]]=1;
        for(int i=2; i<=n; i++)
        {
            Rank[sa[i]]=Rank[sa[i-1]];
            if(a[sa[i]]!=a[sa[i-1]]||b[sa[i]]!=b[sa[i-1]])Rank[sa[i]]++;
        }
    }
    int j=0;
    for(int i=1; i<=n; i++)
    {
        if(j)j--;
        while(ch[i+j-1]==ch[sa[Rank[i]-1]+j-1])j++;
        hight[Rank[i]]=j;
    }
}
int tot,root[maxn];
struct code
{
    int l,r;
    ll sum;
} tree[maxn*80];
void update(int &o,ll L,ll R,ll pos)
{
    tree[++tot]=tree[o];
    o=tot;
    tree[o].sum++;
    if(L==R)
        return ;
    ll mid=(L+R)/2;
    if(pos<=mid)
        update(tree[o].l,L,mid,pos);
    else
        update(tree[o].r,mid+1,R,pos);
}
ll queuy(int s,int e,ll L,ll R,ll l,ll r)
{
    if(L==l&&R==r)
        return tree[e].sum-tree[s].sum;
    ll mid=(L+R)/2;
    if(r<=mid)
        return queuy(tree[s].l,tree[e].l,L,mid,l,r);
    else if(l>mid)
        return queuy(tree[s].r,tree[e].r,mid+1,R,l,r);
    else
        return queuy(tree[s].l,tree[e].l,L,mid,l,mid)+queuy(tree[s].r,tree[e].r,mid+1,R,mid+1,r);
}
ll sum[500005],s,t;
vector<ll>sorted;
int main()
{
    ios::sync_with_stdio(false);
    cin>>n>>s>>t;
    for(int i=0; i<n; i++)
    {
        cin>>ch[i];
        sorted.push_back(ch[i]);
        sum[i+1]=sum[i]+ch[i];
    }
    ll ans=0;
    sort(sorted.begin(),sorted.end());
    for(int i=0; i<n; i++)
        ch[i]=lower_bound(sorted.begin(),sorted.end(),ch[i])-sorted.begin()+1;
    for(int i=1; i<=n; i++)
    {
        root[i]=root[i-1];
        update(root[i],1,4e18,sum[i]+2e18);
    }
    solve();
    for(int i=1; i<=n; i++)
        ans+=queuy(root[sa[i]+hight[i]-1],root[n],1,4e18,sum[sa[i]-1]+s+2e18,sum[sa[i]-1]+t+2e18);
    cout<<ans<<endl;
    return 0;
}