poj3450

题意：求多个字符串的最长公共子串

思路：虽然这道题的最优解不是后缀数组，但是还是学习一下后缀数组的做法，还是和以前一样二分长度，然后再分开计数

[cpp]  view plain  copy

 

1. #include <iostream>  
2. #include <stdio.h>  
3. #include <string.h>  
4. #include <stack>  
5. #include <queue>  
6. #include <map>  
7. #include <set>  
8. #include <vector>  
9. #include <math.h>  
10. #include <bitset>  
11. #include <algorithm>  
12. #include <climits>  
13. using namespace std;  
14.   
15. #define LS 2*i  
16. #define RS 2*i+1  
17. #define UP(i,x,y) for(i=x;i<=y;i++)  
18. #define DOWN(i,x,y) for(i=x;i>=y;i--)  
19. #define MEM(a,x) memset(a,x,sizeof(a))  
20. #define W(a) while(a)  
21. #define gcd(a,b) __gcd(a,b)  
22. #define LL long long  
23. #define N 1000005  
24. #define MOD 1000000007  
25. #define INF 0x3f3f3f3f  
26. #define EXP 1e-8  
27. int wa[N],wb[N],wsf[N],wv[N],sa[N];  
28. int rank[N],height[N],s[N];  
29. //sa:字典序中排第i位的起始位置在str中第sa[i]  
30. //rank:就是str第i个位置的后缀是在字典序排第几  
31. //height：字典序排i和i-1的后缀的最长公共前缀  
32. int cmp(int *r,int a,int b,int k)  
33. {  
34.     return r[a]==r[b]&&r[a+k]==r[b+k];  
35. }  
36. void getsa(int *r,int *sa,int n,int m)//n要包含末尾添加的0  
37. {  
38.     int i,j,p,*x=wa,*y=wb,*t;  
39.     for(i=0; i<m; i++)  wsf[i]=0;  
40.     for(i=0; i<n; i++)  wsf[x[i]=r[i]]++;  
41.     for(i=1; i<m; i++)  wsf[i]+=wsf[i-1];  
42.     for(i=n-1; i>=0; i--)  sa[--wsf[x[i]]]=i;  
43.     p=1;  
44.     j=1;  
45.     for(; p<n; j*=2,m=p)  
46.     {  
47.         for(p=0,i=n-j; i<n; i++)  y[p++]=i;  
48.         for(i=0; i<n; i++)  if(sa[i]>=j)  y[p++]=sa[i]-j;  
49.         for(i=0; i<n; i++)  wv[i]=x[y[i]];  
50.         for(i=0; i<m; i++)  wsf[i]=0;  
51.         for(i=0; i<n; i++)  wsf[wv[i]]++;  
52.         for(i=1; i<m; i++)  wsf[i]+=wsf[i-1];  
53.         for(i=n-1; i>=0; i--)  sa[--wsf[wv[i]]]=y[i];  
54.         t=x;  
55.         x=y;  
56.         y=t;  
57.         x[sa[0]]=0;  
58.         for(p=1,i=1; i<n; i++)  
59.             x[sa[i]]=cmp(y,sa[i-1],sa[i],j)? p-1:p++;  
60.     }  
61. }  
62. void getheight(int *r,int n)//n不保存最后的0  
63. {  
64.     int i,j,k=0;  
65.     for(i=1; i<=n; i++)  rank[sa[i]]=i;  
66.     for(i=0; i<n; i++)  
67.     {  
68.         if(k)  
69.             k--;  
70.         else  
71.             k=0;  
72.         j=sa[rank[i]-1];  
73.         while(r[i+k]==r[j+k])  
74.             k++;  
75.         height[rank[i]]=k;  
76.     }  
77. }  
78.   
79. char str[N],ans[N];  
80. int id[N],vis[4005];  
81.   
82. bool check(int mid,int n,int k)  
83. {  
84.     int i,j,cnt = 0;  
85.     MEM(vis,0);  
86.     for(i = 2; i<=n; i++)  
87.     {  
88.         if(height[i]<mid)  
89.         {  
90.             MEM(vis,0);  
91.             cnt = 0;  
92.             continue;  
93.         }  
94.         if(!vis[id[sa[i-1]]])  
95.         {  
96.             cnt++;  
97.             vis[id[sa[i-1]]] = 1;  
98.         }  
99.         if(!vis[id[sa[i]]])  
100.         {  
101.             cnt++;  
102.             vis[id[sa[i]]] = 1;  
103.         }  
104.         if(cnt == k)  
105.         {  
106.             for(j = 0; j<mid; j++)  
107.                 ans[j] = s[sa[i]+j];  
108.             ans[mid] = '\0';  
109.             return 1;  
110.         }  
111.     }  
112.     return 0;  
113. }  
114.   
115. int main()  
116. {  
117.     int n,i,j,k,len;  
118.     while(~scanf("%d",&k),k)  
119.     {  
120.         n = 0;  
121.         for(i = 0; i<k; i++)  
122.         {  
123.             scanf("%s",str);  
124.             len = strlen(str);  
125.             for(j = 0; j<len; j++)  
126.             {  
127.                 s[n] = str[j];  
128.                 id[n] = i;  
129.                 n++;  
130.             }  
131.             s[n] = '#'+i;  
132.             id[n] = '#'+i;  
133.             n++;  
134.         }  
135.         s[n] = 0;  
136.         getsa(s,sa,n+1,5000);  
137.         getheight(s,n);  
138.         int l = 1,r = len,mid,flag = 0;  
139.         while(l<=r)  
140.         {  
141.             mid = (l+r)/2;  
142.             if(check(mid,n,k))  
143.             {  
144.                 flag = 1;  
145.                 l=mid+1;  
146.             }  
147.             else  
148.                 r=mid-1;  
149.         }  
150.         if(flag)  
151.             printf("%s\n",ans);  
152.         else  
153.             printf("IDENTITY LOST\n");  
154.     }  
155.   
156.     return 0;  
157. }