Common Substrings

后綴數組+單調棧
題解1
題解2
題解3

#include<cstdio> #include<cstring> #include<iostream> using namespace std; typedef long long ll; // sa[i]: 排名是i位的是第幾個后綴 // rk[i]: 第i個后綴的排名是多少 // height[i]: sa[i]與sa[i-1] const int N=200010; char s[N],s1[N],s2[N]; int rk[N],sa[N],cnt[N],height[N]; int x[N],y[N]; int n,m,K; int n1,n2; void rsort()// x[i] 第一關鍵字 y[i] 第二關鍵字 基數排序 {for(int i=1;i<=m;i++) cnt[i]=0;for(int i=1;i<=n;i++) cnt[x[i]]++;for(int i=1;i<=m;i++) cnt[i]+=cnt[i-1];for(int i=n;i;i--) sa[cnt[x[y[i]]]--]=y[i]; } void SA() {n=strlen(s+1);m=300;for(int i=1;i<=n;i++) x[i]=s[i],y[i]=i;rsort();for(int k=1;k<=n;k<<=1){int p=0;for(int i=n-k+1;i<=n;i++) y[++p]=i;// 第二關鍵字為空字符排在最前面for(int i=1;i<=n;i++) if(sa[i]>k) y[++p]=sa[i]-k;rsort();swap(x,y);x[sa[1]]=1,p=1;for(int i=2;i<=n;i++)x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p:++p);if(p==n) break;m=p;}for(int i=1;i<=n;i++) rk[sa[i]]=i;// 求heightfor(int i=1,j=0;i<=n;i++){if(j) --j;while(s[i+j]==s[sa[rk[i]-1]+j]) j++;height[rk[i]]=j;} } int init() {n1=strlen(s1+1);n2=strlen(s2+1);for(int i=1;i<=n1;i++) s[i]=s1[i];s[n1+1]='*';for(int i=1;i<=n2;i++) s[i+n1+1]=s2[i];s[n1+n2+1+1]='\0';return (n1+n2+1); } int st[N][2]; ll solve() {ll ans=0,tot=0;//當前棧里的后綴與將要入棧的后綴的公共字串的個數// A前 B后int tt=0;for(int i=1;i<=n;i++){if(height[i]<K){tt=0,tot=0;continue;}int cnt=0;// 與以后入棧的序列lcp是height[i]的數目if(sa[i-1]<=n1) // 屬于第一個字符串{cnt++;tot+=height[i]-K+1;}while(tt&&height[i]<=st[tt][0]){tot-=1ll*(st[tt][0]-height[i])*st[tt][1];// 由于更小的height[i]導致原先棧中的lcp多算st[tt][0]-height[i]cnt+=st[tt][1];tt--;}st[++tt][0]=height[i];st[tt][1]=cnt;if(sa[i]>n1+1) ans+=tot;// 屬于第二個字符串}// B前 A后tt=0;for(int i=1;i<=n;i++){if(height[i]<K){tt=0,tot=0;continue;}int cnt=0;if(sa[i-1]>n1+1) // 屬于第一個字符串{cnt++;tot+=height[i]-K+1;}while(tt&&height[i]<=st[tt][0]){tot-=1ll*(st[tt][0]-height[i])*st[tt][1];cnt+=st[tt][1];tt--;}st[++tt][0]=height[i];st[tt][1]=cnt;if(sa[i]<=n1+1) ans+=tot;// 屬于第二個字符串}return ans; } int main() {while(scanf("%d",&K),K){scanf("%s%s",s1+1,s2+1);n=init();SA();printf("%lld\n",solve());}return 0; }

總結

以上是生活随笔為你收集整理的POJ - 3415 Common Substrings（长度不小于K的公共子串个数）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。