Codeforces 427D Match & Catch(后缀自动机)
【题目链接】 http://codeforces.com/problemset/problem/427/D
【题目大意】
给出一个两个字符串,求出最短且在两个字符串中唯一的公共子串。
【题解】
以一个字符串建立SAM,Right集大小为1则表明该子串仅出现一次,另一个串在sam上跑,对匹配成功的位置标记+1,当一个位置标记数目为1且right集合大小为1时说明其在两个串中仅出现一次,可以用来更新答案。
【代码】
#include <cstdio> #include <cstring> #include <algorithm> #include <vector> using namespace std; const int N=2000005; char s[N]; struct sam{ int p,q,np,nq,cnt,last,a[N][26],l[N],f[N]; sam(){cnt=0;last=++cnt;} void extend(int c){ p=last;np=last=++cnt;l[np]=l[p]+1; while(!a[p][c]&&p)a[p][c]=np,p=f[p]; if(!p)f[np]=1; else{ q=a[p][c]; if(l[p]+1==l[q])f[np]=q; else{ nq=++cnt;l[nq]=l[p]+1; memcpy(a[nq],a[q],sizeof(a[q])); f[nq]=f[q]; f[np]=f[q]=nq; while(a[p][c]==q)a[p][c]=nq,p=f[p]; } } }int b[N],x[N],r[N]; void build(){ scanf("%s",s+1); int len=strlen(s+1); for(int i=1;i<=len;i++)extend(s[i]-'a'); for(int i=1;i<=cnt;i++)b[l[i]]++; for(int i=1;i<=len;i++)b[i]+=b[i-1]; for(int i=1;i<=cnt;i++)x[b[l[i]]--]=i; for(int i=p=1;i<=len;i++){p=a[p][s[i]-'a'];r[p]++;} for(int i=cnt;i;i--)r[f[x[i]]]+=r[x[i]]; }int u[N]; void solve(){ scanf("%s",s+1); int p=1,len=strlen(s+1),now=1,tmp=0,ans=len+1; for(int i=1;i<=len;i++){ int c=s[i]-'a'; if(a[p][c])p=a[p][c],u[p]++; else{ while(p&&!a[p][c])p=f[p]; if(!p)p=1; else p=a[p][c],u[p]++; } }for(int i=cnt;i;i--)u[f[x[i]]]+=u[x[i]]; for(int i=1;i<=cnt;i++)if(u[i]==1&&r[i]==1)ans=min(ans,l[f[i]]+1); if(ans==len+1)puts("-1"); else printf("%d\n",ans); } }sam; int main(){ sam.build(); sam.solve(); return 0; }