正题
题目大意
一个字符串SSS。
若干个询问,每次询问一个串TTT和l,rl,rl,r。询问有多少个TTT和SSS的公共子串满足和为[l,r][l,r][l,r]
解题思路
考虑枚举子串左端,那么右串一定在一个范围内,考虑如何求出一个范围。
考虑用后缀数组解决这个问题,我们把所有串拼在一起。对与每个在TTT里的后缀,我们要找到左右边的第一个SSS里的后缀,然后用树状数组维护答案。
时间复杂度O(nlogn)O(n\log n)O(nlogn)
codecodecode
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define mp(x,y) make_pair(x,y)
#define lowbit(x) (x&-x)
using namespace std;
const int N=6e5+10,lim=2e6;
int n,m,q,s[N],l[N],height[N],L[N],R[N];
int x[N],y[N],sa[N],rk[N],c[N];
pair<int,int> p[N];
vector<int> v[N],wz[N];
char st[N];
struct Tree_Array{int t[lim+10];void Change(int x,int val){x++;while(x<=lim){t[x]+=val;x+=lowbit(x);}return;}int Ask(int x){int ans=0;x++;if(x<0)return 0;while(x){ans+=t[x];x-=lowbit(x);}return ans;}
}T;
void Init(){scanf("%s",st+1);n=strlen(st+1);for(int i=1;i<=n;i++)s[i]=st[i]-'0'+1;scanf("%d",&q);for(int i=1;i<=q;i++){scanf("%s",st+1);l[i]=strlen(st+1);s[++n]=11;for(int j=1;j<=l[i];j++){s[++n]=st[j]-'0'+1,p[n]=mp(i,j);v[i].push_back(-1); wz[i].push_back(n);}scanf("%d%d",&L[i],&R[i]);}
}
void Qsort(){for(int i=1;i<=m;i++)c[i]=0;for(int i=1;i<=n;i++)c[x[i]]++;for(int i=1;i<=m;i++)c[i]+=c[i-1];for(int i=n;i>=1;i--)sa[c[x[y[i]]]--]=y[i],y[i]=0;
}
void Get_SA(){m=11;for(int i=1;i<=n;i++)x[i]=s[i],y[i]=i;Qsort();for(int w=1;w<=n;w<<=1){int p=0;for(int i=n-w+1;i<=n;i++)y[++p]=i;for(int i=1;i<=n;i++)if(sa[i]>w)y[++p]=sa[i]-w;Qsort();swap(x,y);x[sa[1]]=p=1;for(int i=2;i<=n;i++)x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+w]==y[sa[i-1]+w])?p:++p;if(p==n)break;m=p;}return;
}
void Get_Height(){int k=0;for(int i=1;i<=n;i++)rk[sa[i]]=i;for(int i=1;i<=n;i++){if(rk[i]==1)continue;if(k)k--;int j=sa[rk[i]-1];while(i+k<=n&&j+k<=n&&s[i+k]==s[j+k])++k;height[rk[i]]=k;}return;
}
void Solve(){int w=2147483647/3,flag=0;for(int i=1;i<=n;i++){int x=sa[i];if(!p[x].first)w=2147483647/3,flag=1;else{if(flag){w=min(w,height[i]);v[p[x].first][p[x].second]=w;}}}w=2147483647/3;flag=0;for(int i=n;i>=1;i--){int x=sa[i]; if(!p[x].first)w=height[i],flag=1;else{if(flag){v[p[x].first][p[x].second]=max(v[p[x].first][p[x].second],w);w=min(w,height[i]);}}}return;
}
void Get_Ans(){for(int x=1;x<=q;x++){T.Change(0,1);int sum=0,k=0,r=l[x];long long ans=0;for(int i=l[x];i>=1;i--){sum+=s[wz[x][i-1]]-1;while(r&&r>=i+v[x][i]){r--;T.Change(k,-1);k+=s[wz[x][r]]-1;}ans+=T.Ask(sum-L[x])-T.Ask(sum-R[x]-1);T.Change(sum,1);}while(r){r--;T.Change(k,-1);k+=s[wz[x][r]]-1;}T.Change(k,-1);printf("%lld\n",ans);}return;
}
int main()
{Init();Get_SA();Get_Height();Solve();Get_Ans();
}