poj 1743 后缀数组

 

思路：要求A中两子串之差为定值，将每相邻两个数相减所得数列B,等价于要求B中子串相等。从而转换LCP(longest commen prefix)问题，又要求不能重叠，采用二分答案方法求解。要注意的是二分期间的判断search_ans：若果h[i]>=len，并且h[j]>=len，则sa[i]..s[j]中的最大与最小值之差即为重叠瓶颈！

 

1. #include <iostream>
2. #include <string>
3. #include <algorithm>
4. using namespace std;
6. #define Min(a,b) (a)<(b)?(a):(b)
7. #define Max(a,b) (a)>(b)?(a):(b)
8. const int N = 41000;
9. const int D = 10;
10. const int M = 200, offset = 100;
11. int n;
12. int s[N];
13. int cnt[N], mem[4][N], *rank, *nrank, *sa, *nsa, h[N];
14. // lcp[i][j]: longest commen prefix ( suffix(sa[k+1]), suffix(sa[k]) ) j <= k < j+2^i
16. void radix_sort()
17. {
18.  int i, j, k;
19.  rank = mem[0];
20.  nrank = mem[1];
21.  sa = mem[2];
22.  nsa = mem[3];
24.  for(i = 0; i < n; i++) cnt[s[i]]++;
25.  for(i = 1; i < M; i++) cnt[i] += cnt[i-1];
26.  for(i = n-1; i >= 0; i--) sa[--cnt[s[i]]] = i;
27.  for(rank[0]=0, i=1; i < n; i++)
28.  {
29.   rank[sa[i]] = rank[sa[i-1]];
30.   if(s[sa[i]]!=s[sa[i-1]]) rank[sa[i]]++;
31.  }
33.  for(k = 1; k<n && rank[sa[n-1]] < n-1; k*=2)
34.  {
35.   for(i = 0; i < n; i++) cnt[rank[sa[i]]] = i+1;
36.   for(i = n-1; i >= 0; i--) if(sa[i]-k>=0)
37.    nsa[--cnt[rank[sa[i]-k]]] = sa[i]-k;
39.   // max(sa[i]-k)=n-k-1 , therefore i = n-k;
40.   for(i = n-k; i < n; i++)
41.    nsa[--cnt[rank[i]]] = i;
43.   for(nrank[nsa[0]], i=1; i < n; i++)
44.   {
45.    nrank[nsa[i]] = nrank[nsa[i-1]];
46.    if(rank[nsa[i]] != rank[nsa[i-1]]
47.     || rank[nsa[i]+k] != rank[nsa[i-1]+k])
48.    nrank[nsa[i]]++;
49.   }
50.   swap(rank, nrank);
51.   swap(sa, nsa);
52.  }
53. }
57. void get_lcp_rmq()
58. {
59.  int i, j, k;
60.  for(i=0,k=0; i<n; i++)
61.  {
62.   if(rank[i]==n-1) h[rank[i]]=k=0;
63.   else
64.   {
65.    if(k>0)k--;
66.    j = sa[rank[i]+1];
67.    for(;s[i+k]==s[j+k];k++) ;
68.    h[rank[i]]=k;
69.   }
70.  }
73. }
76. /*
77. 9
78. 1 2 3 4 5 6 7 8 9
80. */
82. bool search_ans(int len)
83. {
84.  int i, l=N, r=0, ma, mi;
86.  for(i = 0; i < n-2; i++)
87.  {
88.   if( h[i] < len) l=N,r=0;
89.   else
90.   {
91.    mi = sa[i];
92.    ma = sa[i+1];
93.    if(mi > ma) swap(mi,ma);
94.    l = Min(mi, l);
95.    r = Max(ma, r);
96.    if(r - l >= len)return true;
97.   }
98.  }
99.  return false;
100. }
103. int main()
104. {
105.  int i, j, k;
109.  while(scanf("%d", &n)==1 && n)
110.  {
111.   memset(cnt, 0, sizeof(cnt));
112.   memset(mem, 0, sizeof(mem));
113.   for(i=0;i<n;i++) scanf("%d",&s[i]);
115.   for(i=0;i<n-1;i++) s[i] = s[i+1]-s[i]+offset;
116.   s[n-1]=0;
118.   //for(i=0;i<n;i++)printf("%d ",s[i]); puts("");
119.   radix_sort();
121.   get_lcp_rmq();
123.   int low = 0, up = n, mid;
124.   bool get=0;
125.   while(low < up)
126.   {
127.    mid = (low+up+1)/2;
128.    if( get=search_ans(mid) ) low = mid;
129.    else up = mid-1;
130.   }
132.   if(low < 4)puts("0");
133.   else printf("%d/n", low+1);
134.  }
136.     return 0;
137. }