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;
}