正題

---

題目大意

一個字符串$S$。

若干個詢問，每次詢問一個串$T$和$l,r$。詢問有多少個$T$和$S$的公共子串滿足和為$[l,r]$

---

解題思路

考慮枚舉子串左端，那么右串一定在一個范圍內(nèi)，考慮如何求出一個范圍。

考慮用后綴數(shù)組解決這個問題，我們把所有串拼在一起。對與每個在$T$里的后綴，我們要找到左右邊的第一個$S$里的后綴，然后用樹狀數(shù)組維護答案。

時間復(fù)雜度$O(n\log n)$

---

$code$

#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(); } 創(chuàng)作挑戰(zhàn)賽新人創(chuàng)作獎勵來咯，堅持創(chuàng)作打卡瓜分現(xiàn)金大獎

總結(jié)

以上是生活随笔為你收集整理的[2020.10.30NOIP模拟赛]字符串水题【SA,树状数组】的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。