POJ 2774 后缀数组

题目要求：求s1,s2的最大子串

思路：将s1,s2合并为一个字符串s, 也就是求s的max(lcp[i][j]), 唯一i,j分别位于s1和s2, 利用后缀数组计算s的h[]或者height[]数组,那么答案即为h中的最大值。

证明如下：s1,s2一定存在最大子串t，t为s1的子串t1和s2的子串t2的lcp[t1][t2], 假设t1,t2在后缀数组中不相邻，则任意取后缀数组中位于t1,t2之间的串tt, 则lcp[tt][t1]与lcp[tt][t2]中至少有一个同时满足

    (1) >=lcp[t1][t2] ：这是由后缀数组的性质所决定的

    (2) 两串分别位于s1,s2：这是因为t1,t2分别属于s1,s2

从而得到更优的解，所以答案一定为h中的最大值

2. #include <iostream>
3. #include <string>
4. #include <algorithm>
5. using namespace std;
7. #define Min(a,b) (a)<(b)?(a):(b)
8. const int N = 201000;
9. int n, m;
10. char s[N], s2[N];
12. int cnt[N], mem[4][N], *rank, *nrank, *sa, *nsa, h[N];
13. // lcp[i][j]: longest commen prefix ( suffix(sa[k+1]), suffix(sa[k]) ) j <= k < j+2^i
15. void radix_sort()
16. {
17.     int i, j, k;
18.     rank = mem[0];
19.     nrank = mem[1];
20.     sa = mem[2];
21.     nsa = mem[3];
23.     for(i = 0; i < n; i++) cnt[s[i]]++;
24.     for(i = 1; i < 256; i++) cnt[i] += cnt[i-1];
25.     for(i = n-1; i >= 0; i--) sa[--cnt[s[i]]] = i;
26.     for(rank[0]=0, i=1; i < n; i++)
27.     {
28.         rank[sa[i]] = rank[sa[i-1]];
29.         if(s[sa[i]]!=s[sa[i-1]]) rank[sa[i]]++;
30.     }
32.     for(k = 1; k<n && rank[sa[n-1]] < n-1; k*=2)
33.     {
34.         for(i = 0; i < n; i++) cnt[rank[sa[i]]] = i+1;
35.         for(i = n-1; i >= 0; i--) if(sa[i]-k>=0)
36.             nsa[--cnt[rank[sa[i]-k]]] = sa[i]-k;
38.         // max(sa[i]-k)=n-k-1 , therefore i = n-k;
39.         for(i = n-k; i < n; i++)
40.             nsa[--cnt[rank[i]]] = i;
42.         for(nrank[nsa[0]], i=1; i < n; i++)
43.         {
44.             nrank[nsa[i]] = nrank[nsa[i-1]];
45.             if(rank[nsa[i]] != rank[nsa[i-1]]
46.                 || rank[nsa[i]+k] != rank[nsa[i-1]+k])
47.             nrank[nsa[i]]++;
48.         }
49.         swap(rank, nrank);
50.         swap(sa, nsa);
51.     }
52. }
55. void get_lcp_rmq()
56. {
57.     int i, j, k;
58.     for(i=0,k=0; i<n; i++)
59.     {
60.         if(rank[i]==n-1) h[rank[i]]=k=0;
61.         else
62.         {
63.             if(k>0)k--;
64.             j = sa[rank[i]+1];
65.             for(;s[i+k]==s[j+k];k++) ;
66.             h[rank[i]]=k;
67.         }
68.     }
70. }
75. int main()
76. {
77.     int i, j, k;
78.     int p1, p2, n1;
80.     gets(s);
81.     n1 = strlen(s);
82.     s[n1++]='#';
84.     gets(s2);
85.     strcat(s,s2);
86.     n = strlen(s);
87.     s[n++]=0;
89.     radix_sort();
90.     get_lcp_rmq();
92.     int ans = 0;
93.     for(i = 0; i < n-1; i++)
94.     {
95.         j = sa[i];
96.         if(j < n1)p1 = 1;
97.         else p1 = -1;
99.         k = sa[i+1];
100.         if(k < n1)p2 = 1;
101.         else p2 = -1;
103.         if(p1*p2<1 && h[i]>ans)
104.             ans = h[i];
105.     }
106.     printf("%d/n", ans);
107.     return 0;
108. }