PKU 2774 Long Long Message 后缀数组 最长公共子串

给两个字符串，求他们的最长公共子串。

把他们连接起来，就是求后缀数组，找最大的height，注意i和i-1要分别在两边

1. #include <iostream>  
2. #include <stdio.h>  
3. #include <string.h>  
4. #include <algorithm>  
5. using namespace std;  
6. #define M 303  
7. #define inf 0x3fffffff  
8. #define maxn 500000  
9. #define ws ww  
10. #define rank RANK  
11. #define F(x) ((x)/3+((x)%3==1?0:tb))  
12. #define G(x) ((x)<tb?(x)*3+1:((x)-tb)*3+2)   
13. int wa[maxn],wb[maxn],wv[maxn],ws[maxn];  
14. int r[maxn],sa[maxn];  
15. int cmp(int *r,int a,int b,int l)  
16. {  
17.     return r[a]==r[b]&&r[a+l]==r[b+l];  
18. }  
19. void da(int *r,int *sa,int n,int m)  
20. {  
21.     int i,j,p,*x=wa,*y=wb,*t;   
22.     for(i=0;i<m;i++) ws[i]=0;   
23.     for(i=0;i<n;i++) ws[x[i]=r[i]]++;   
24.     for(i=1;i<m;i++) ws[i]+=ws[i-1];   
25.     for(i=n-1;i>=0;i--) sa[--ws[x[i]]]=i;   
26.     for(j=1,p=1;p<n;j*=2,m=p)   
27.     {   
28.         for(p=0,i=n-j;i<n;i++) y[p++]=i;   
29.         for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;   
30.         for(i=0;i<n;i++) wv[i]=x[y[i]];   
31.         for(i=0;i<m;i++) ws[i]=0;   
32.         for(i=0;i<n;i++) ws[wv[i]]++;   
33.         for(i=1;i<m;i++) ws[i]+=ws[i-1];   
34.         for(i=n-1;i>=0;i--) sa[--ws[wv[i]]]=y[i];   
35.         for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)   
36.             x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;   
37.     }   
38.     return;   
39. }  
40. int rank[maxn],height[maxn];  
41. void calheight(int *r,int *sa,int n)  
42. {  
43.     int i,j,k=0;  
44.     for(i=1;i<=n;i++)   
45.         rank[sa[i]]=i;  
46.     for(i=0;i<n;height[rank[i++]]=k)  
47.         for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);  
48.     return;  
49. }  
50. char str[maxn];  
51. int main()  
52. {  
53.     gets(str);  
54.     int len1;  
55.     len1=strlen(str);  
56.     str[len1]='$';  
57.     gets(str+len1+1);  
58.     int len=strlen(str);  
59.     for(int i=0;i<len;i++)  
60.         r[i]=str[i];  
61.     da(r,sa,len+1,128);  
62.     calheight(r,sa,len);  
63.     int ans=0;  
64.     for(int i=1;i<len;i++)  
65.     {  
66.         if((sa[i-1]<len1&&sa[i]>=len1)||(sa[i-1]>=len1&&sa[i]<len1))  
67.             ans=max(ans,height[i]);  
68.     }  
69.     printf("%d\n",ans);  
70.     return 0;  
71. }