PKU 1743 Musical Theme 解题报告

果然是楼天成男人8题之一。。。。

1. /*
2.   Title: Musical Theme
3.   Author: Jeff
4.   Time:  2008/10/31
5.   Complexity: O(N*log(N));
6.   Result: 1100K 516MS/ AC
7.   Reference: 
8.         http://blog.csdn.net/milan_2008/archive/2008/09/15/2932293.aspx
9.         http://203.208.39.99/search?q=cache:K84ULK6d-WoJ:hi.baidu.com/zfy0701/blog/item/f2278a0928991dca3bc763a0.html+pku+1743&hl=zh-CN&lr=lang_zh-CN|lang_zh-TW&newwindow=1&gl=cn&st_usg=ALhdy29Gp86u2sgbCS51uzMjjLxHnuq5IA&strip=1
10.   Description: 
11.         求最长不重叠差为定值子串的长度
12.   Tips:
13.         LCP问题，后缀数组。 
14.         对于这道，要先把串转换一下，根据题目的性质，应该把输入得到的
15.         串前后相减得到方便求解的新的串——设其为s，再求该串s中最长的
16.         不重叠重复子串。由于不能重叠，导致height数组的最大值不一定是
17.         解，因为相邻两串可能会重叠。
19.         此题用后缀数组也有两种解法，1：二分枚举答案，2：用栈线性扫描，
20.         主要说做法1，因为它更以理解些
22.         现在我们假设-最长,不重叠,重复-子串的长度为len，那么在一个
23.         height数组中有一些hgt[i]会小于len，在这个i左右的两个子串，他
24.         们LCP是不可能大于或等于len的，这样，就可以吧height数组看做很
25.         多LCP >= len的段，我们在每一段中进行扫描，记录这一段中最大和
26.         最小的子串串索引（sa[x]），如果两者之和小于len，说明重叠了，
27.         否则就找到了一个可行解。
29.         易证：如果该串存在len1的不重叠子串，且len1 > len2，则该串也存
30.         在长度len2的不重叠子串，解有连续性，所以我们在上面这个过程外我
31.         们可以二分枚举解。
33.         对了，二分查找之前之前可先调用查找len = 4是否成立，不成立就直
34.         接return了。
36.         这道题的关键点就是：判断解的时候对height数组进行分段处理，这个
37.         分段的思想与其说重要，还不说就是height数组的基本性质。类似的分
38.         段法在下题中再次用到。
39.   p.s. 
40. */
41. #include <cstdio>
42. #include <cstring>
43. #include <algorithm>
45. using namespace std;
47. const int MAX = 20001;
49. struct Sfx{
50.     int idx;
51.     int key[2];
52.     bool operator<(const Sfx &other)const{
53.         return (key[0] < other.key[0]) ||
54.             (key[0] == other.key[0] && key[1] < other.key[1]);    
55.     }    
56. };
58. int s[MAX];
59. Sfx sa[MAX], temp_sa[MAX];
60. int rank[MAX];
61. int h[MAX];
62. int hgt[MAX];
64. inline int max(int a, int b){
65.     return a > b ? a : b;    
66. }
67. inline int min(int a, int b){
68.     return a < b ? a : b;
69. }
71. void cSort(Sfx *in, int key, int len, Sfx *out){
72.     memset(h, 0, sizeof(h));
73.     int *cnt = h;
74.     for(int i = 0; i < len; i++)cnt[ in[i].key[key] ]++;
75.     for(int i = 1; i < len; i++)cnt[i] += cnt[i-1];
76.     for(int i = len - 1; i >= 0; i--)
77.         out[ --cnt[in[i].key[key]] ] = in[i];
78. }
80. void build_sfx(int len){
81.     for(int i = 0; i < len; i++){
82.         sa[i].idx = sa[i].key[1] = i;
83.         sa[i].key[0] = s[i];    
84.     }    
85.     sort(sa, sa + len);
86.     for(int i = 0; i < len; i++)sa[i].key[1] = 0;
87.     
88.     int wid = 1;
89.     while(wid < len){
90.         //为0的时候，因为当所有的数都相同时，比较不出大小。 
91.         //所以在所有数后面加了一个无穷小的数。 
92.         rank[sa[0].idx] = 0;    
93.         for(int i = 1; i < len; i++){
94.             rank[sa[i].idx] = rank[sa[i-1].idx];
95.             if(sa[i-1] < sa[i])
96.                 rank[sa[i].idx]++;    
97.         }
98.         //printf("r: ");for(int i = 0; i < len; i++)printf("%d ", rank[i]);printf("/n");
99.         for(int i = 0; i < len; i++){
100.             sa[i].idx = i;
101.             sa[i].key[0] = rank[i];    
102.             //key[1]不能为-1, 因为cSort对大于等于0的数进行统计。 
103.             sa[i].key[1] = (i + wid < len ? rank[i+wid] : 0); 
104.         }
105.         cSort(sa, 1, len, temp_sa); cSort(temp_sa, 0, len, sa);
106.         //printf("sa:");for(int i = 0; i < len; i++)printf("%d ", sa[i].idx);printf("/n");
107.         wid *= 2;
108.     }
109.     //需要再计算一次rank值，因为以上最后一次cSort后，rank的计算还差一次。 
110.     rank[sa[0].idx] = 0;    
111.     for(int i = 1; i < len; i++){
112.         rank[sa[i].idx] = rank[sa[i-1].idx];
113.         if(sa[i-1] < sa[i])rank[sa[i].idx]++;    
114.     }
115. }
117. void calH(int len){
118.     for(int i = 0; i < len; i++){
119.         if(rank[i] == 0){
120.             h[i] = 0;
121.             hgt[rank[i]] = 0;
122.         }
123.         else{
124.             int k;
125.             if(i == 0 || h[i-1] < 1)k = 0;
126.             else k = h[i-1] - 1;
127.             for(int j = sa[rank[i]-1].idx; s[j + k] == s[i + k]; k++);
128.             h[i] = k;
129.             hgt[rank[i]] = k;
130.         }    
131.     }    
132. }
134. bool check(int mid, int len){
135.     int ma = 0, mi = MAX;
136.     for(int i = 1; i < len; i++){
137.         if(hgt[i] < mid){
138.             ma = 0;
139.             mi = MAX;
140.         }else{
141.             int minl = min(sa[i].idx, sa[i-1].idx);
142.             int maxr = max(sa[i].idx, sa[i-1].idx);
143.             mi = min(minl, mi);
144.             ma = max(maxr, ma);
145.             if(ma - mi >= mid)return true;    
146.         }
147.     }   
148.     return false;
149. }
151. int main(){
152.     freopen("in.txt", "r", stdin);
153.     freopen("out.txt", "w", stdout);
154.     int N;
155.     while(scanf("%d", &N) && N != 0){
156.         int now, pre;
157.         scanf("%d", &pre);
158.         for(int i = 0; i < N - 1; i++){
159.             scanf("%d", &now);
160.             s[i] = now - pre + 100; //计算前后数值的差，最后求其实就是长度>=4 
161.             pre = now;    
162.         }
163.         //末尾结束符 
164.         s[N-1] = -MAX;
165.         build_sfx(N);
166.         calH(N);
167.         //printf("r: ");for(int i = 0; i < N; i++)printf("%d ", rank[i]);printf("/n");
168.         //printf("sa:");for(int i = 0; i < N; i++)printf("%d ", sa[i].idx);printf("/n");
169.         //printf("h: ");for(int i = 0; i < N; i++)printf("%d ", h[i]);printf("/n");
170.         //printf("hgt:");for(int i = 0; i < N; i++)printf("%d ", hgt[i]);printf("/n");
171.         int ans = -1;
172.         int l = 4, r = N/2+1;
173.         int mid;
174.         while(l <= r){
175.             mid = (l + r) / 2;
176.             if(check(mid, N-1)){
177.                 l = mid + 1;
178.                 ans = mid;
179.             }    
180.             else
181.                 r = mid - 1;
182.         }
183.         printf("%d/n", ans+1);
184.     }
185.     return 0;    
186. }