题目描述:
给定两个字符串,求出长度不小于k的公共字串的个数。
算法标签:后缀数组
以下代码:
![](https://i-blog.csdnimg.cn/blog_migrate/8f900a89c6347c561fdf2122f13be562.gif)
![](https://i-blog.csdnimg.cn/blog_migrate/961ddebeb323a10fe0623af514929fc1.gif)
#include<bits/stdc++.h> #define il inline #define LL long long #define _(d) while(d(isdigit(ch=getchar()))) using namespace std; const int N=2e5+5; char s[N],b[N];int l1,l2,n,k,sa[N],rk[N],height[N],x[N],y[N],c[N],top,q[N][2];LL ans,tot; il int read(){int x,f=1;char ch;_(!)ch=='-'?f=-1:f;x=ch^48;_()x=(x<<1)+(x<<3)+(ch^48);return f*x;} il void getsa(){ int m=130; for(int i=1;i<=m;i++)c[i]=0; for(int i=1;i<=n;i++)c[x[i]=s[i]]++; for(int i=2;i<=m;i++)c[i]+=c[i-1]; for(int i=n;i;i--)sa[c[x[i]]--]=i; for(int g=1;g<=n;g<<=1){ int num=0; for(int i=n-g+1;i<=n;i++)y[++num]=i; for(int i=1;i<=n;i++)if(sa[i]>g)y[++num]=sa[i]-g; for(int i=1;i<=m;i++)c[i]=0; for(int i=1;i<=n;i++)c[x[i]]++; for(int i=2;i<=m;i++)c[i]+=c[i-1]; for(int i=n;i;i--)sa[c[x[y[i]]]--]=y[i],y[i]=0; swap(x,y);num=1;x[sa[1]]=1; for(int i=2;i<=n;i++) x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+g]==y[sa[i-1]+g])?num:++num; if(num==n)break;m=num; } for(int i=1;i<=n;i++)rk[sa[i]]=i; } il void geth(){ int g=0; for(int i=1;i<=n;i++){ if(rk[i]==1)continue; if(g)g--;int j=sa[rk[i]-1]; while(i+g<=n&&j+g<=n&&s[i+g]==s[j+g])g++; height[rk[i]]=g; } } int main() { while(1){ k=read();if(k==0)break; scanf(" %s",s+1);n=l1=strlen(s+1);scanf(" %s",b+1);l2=strlen(b+1); s[++n]='@';for(int i=1;i<=l2;i++)s[++n]=b[i];s[++n]='0'; getsa();geth(); top=tot=ans=0; for(int i=2;i<=n;i++){ if(height[i]<k){tot=top=0;continue;} int cnt=0; if(sa[i-1]<=l1)cnt++,tot+=height[i]-k+1; while(top>0&&height[i]<=q[top][0]){ tot-=q[top][1]*(q[top][0]-height[i]); cnt+=q[top--][1]; } q[++top][0]=height[i];q[top][1]=cnt; if(sa[i]>l1+1)ans+=tot; } tot=top=0; for(int i=2;i<=n;i++){ if(height[i]<k){tot=top=0;continue;} int cnt=0; if(sa[i-1]>l1+1)cnt++,tot+=height[i]-k+1; while(top>0&&height[i]<=q[top][0]){ tot-=q[top][1]*(q[top][0]-height[i]); cnt+=q[top--][1]; } q[++top][0]=height[i];q[top][1]=cnt; if(sa[i]<=l1)ans+=tot; } printf("%lld\n",ans); } return 0; }