POJ-2774 Long Long Message 后缀数组
题目链接：http://poj.org/problem?id=2774
两个字符串的最长公共字串。
求出height数组后直接二分答案就可以了，或者线性扫描一遍。
1 //STATUS:C++_AC_594MS_4800KB 2 #include<stdio.h> 3 #include<stdlib.h> 4 #include<string.h> 5 #include<math.h> 6 #include<iostream> 7 #include<string> 8 #include<algorithm> 9 #include<vector> 10 #include<queue> 11 #include<stack> 12 #include<map> 13 using namespace std; 14 #define LL __int64 15 #define pii pair<int,int> 16 #define mem(a,b) memset(a,b,sizeof(a)) 17 #define lson l,mid,rt<<1 18 #define rson mid+1,r,rt<<1|1 19 #define PI acos(-1.0) 20 const int N=200010,INF=0x3f3f3f3f,MOD=10000,STA=8000010; 21 //const LL LNF=0x3f3f3f3f3f3f3f3f; 22 const double DNF=1e13; 23 // 24 inline int Max(int a,int b){return a>b?a:b;} 25 inline int Min(int a,int b){return a<b?a:b;} 26 void swap(int& a,int& b){int t=a;a=b;b=t;} 27 void swap(LL& a,LL& b){LL t=a;a=b;b=t;} 28 // 29 30 char s[N/2]; 31 int num[N]; 32 int sa[N],t1[N],t2[N],c[N],rank[N],height[N]; 33 int n,m,len1,len2; 34 35 void build_sa(int s[],int n,int m) 36 { 37 int i,k,p,*x=t1,*y=t2; 38 //第一轮基数排序 39 for(i=0;i<m;i++)c[i]=0; 40 for(i=0;i<n;i++)c[x[i]=s[i]]++; 41 for(i=1;i<m;i++)c[i]+=c[i-1]; 42 for(i=n-1;i>=0;i--)sa[--c[x[i]]]=i; 43 for(k=1;k<=n;k<<=1){ 44 p=0; 45 //直接利用sa数组排序第二关键字 46 for(i=n-k;i<n;i++)y[p++]=i; 47 for(i=0;i<n;i++)if(sa[i]>=k)y[p++]=sa[i]-k; 48 //基数排序第一关键字 49 for(i=0;i<m;i++)c[i]=0; 50 for(i=0;i<n;i++)c[x[y[i]]]++; 51 for(i=1;i<m;i++)c[i]+=c[i-1]; 52 for(i=n-1;i>=0;i--)sa[--c[x[y[i]]]]=y[i]; 53 //根据sa和x数组计算新的x数组 54 swap(x,y); 55 p=1;x[sa[0]]=0; 56 for(i=1;i<n;i++) 57 x[sa[i]]=y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++; 58 if(p>=n)break; //已经排好序，直接退出 59 m=p; //下次基数排序的最大值 60 } 61 } 62 63 void getHeight(int s[],int n) 64 { 65 int i,j,k=0; 66 for(i=0;i<=n;i++)rank[sa[i]]=i; 67 for(i=0;i<n;i++){ 68 if(k)k--; 69 j=sa[rank[i]-1]; 70 while(s[i+k]==s[j+k])k++; 71 height[rank[i]]=k; 72 } 73 } 74 75 int binary(int l,int r) 76 { 77 int i,j,mid,ok1,ok2,ret; 78 while(l<=r){ 79 mid=(l+r)>>1; 80 ok1=ok2=0; 81 if(sa[1]<len1)ok1=1; 82 else if(sa[1]>len1)ok2=1; 83 for(i=2;i<=n;i++){ 84 if(height[i]<mid)ok1=ok2=0; 85 if(sa[i]<len1)ok1=1; 86 else if(sa[i]>len1)ok2=1; 87 if(ok1 && ok2)break; 88 } 89 if(ok1 && ok2)ret=mid,l=mid+1; 90 else r=mid-1; 91 } 92 return ret; 93 } 94 95 int main() 96 { 97 // freopen("in.txt","r",stdin); 98 int i,j; 99 while(~scanf("%s",s) && s[0]!='#') 100 { 101 len1=strlen(s); 102 for(i=0;i<len1;i++)num[i]=s[i]-'a'+1; 103 num[len1]=27; 104 scanf("%s",s); 105 len2=strlen(s); 106 n=len1+len2+1; 107 for(i++,j=0;i<n;i++,j++)num[i]=s[j]-'a'+1; 108 num[n]=0; 109 m=28; 110 build_sa(num,n+1,m); 111 getHeight(num,n); 112 113 printf("%d\n",binary(0,Min(len1,len2))); 114 } 115 return 0; 116 }
