spoj220

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 100005  
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],a[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];  
80. int id[N],maxn[N],minn[N];  
81.   
82. bool check(int mid,int n,int k)  
83. {  
84.     int i,j;  
85.     for(i = 0; i<=k; i++)  
86.     {  
87.         maxn[i] = 0;  
88.         minn[i] = INF;  
89.     }  
90.     for(i = 1; i<=n; i++)  
91.     {  
92.         if(height[i]<mid)  
93.         {  
94.             for(j = 0; j<=k; j++)  
95.             {  
96.                 maxn[j] = 0;  
97.                 minn[j] = INF;  
98.             }  
99.             maxn[id[sa[i]]] = sa[i];  
100.             minn[id[sa[i]]] = sa[i];  
101.         }  
102.         else  
103.         {  
104.             //求出第id[sa[i]]个串中，所使用头和尾，差值必须大于等于二分的答案  
105.             maxn[id[sa[i]]] = max(maxn[id[sa[i]]],sa[i]);  
106.             minn[id[sa[i]]] = min(minn[id[sa[i]]],sa[i]);  
107.             maxn[id[sa[i-1]]] = max(maxn[id[sa[i-1]]],sa[i-1]);  
108.             minn[id[sa[i-1]]] = min(minn[id[sa[i-1]]],sa[i-1]);  
109.             for(j = 0; j<k; j++)  
110.             {  
111.                 if(maxn[j]-minn[j]<mid)  
112.                     break;  
113.             }  
114.             if(j==k) return true;  
115.         }  
116.     }  
117.     return false;  
118. }  
119.   
120. int main()  
121. {  
122.     int t,n,i,j,k;  
123.     scanf("%d",&t);  
124.     while(t--)  
125.     {  
126.         scanf("%d",&k);  
127.         n = 0;  
128.         for(i = 0; i<k; i++)  
129.         {  
130.             scanf("%s",str+n);  
131.             for(; str[n]!='\0'; n++)  
132.             {  
133.                 s[n] = str[n];  
134.                 id[n] = i;  
135.             }  
136.             s[n++] = '#'+i;  
137.         }  
138.         s[n-1] = 0;  
139.         getsa(s,sa,n,255);  
140.         getheight(s,n-1);  
141.         int l = 0,r = 10000,mid,ans = 0;  
142.         while(l<=r)  
143.         {  
144.             mid = (l+r)/2;  
145.             if(check(mid,n,k))  
146.             {  
147.                 ans = mid;  
148.                 l = mid+1;  
149.             }  
150.             else r = mid-1;  
151.         }  
152.         printf("%d\n",ans);  
153.     }  
154.   
155.     return 0;  
156. }  //似乎是poj3294变形？