后缀数组 例题后缀排序 uoj #35

 

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 int const N = 300000 + 3;
 4 int sa[N], rk[N], num[N], r[N], a[N], wa[N], wb[N], wv[N], n, m,h[N];  
 5 char s1[N], s[N];
 6 int cmp(int *r, int a, int b, int l) {
 7     return r[a] == r[b] &&  r[a + l] == r[b + l];
 8 }
 9 void build_sa(char *r, int *sa, int n, int m) {
10     int i, j, p, *x = wa, *y = wb;
11     for(i = 0; i < m; i++)
12         num[i] = 0;
13     for(i = 0; i < n; i++)
14         num[x[i] = r[i]]++;
15     for(i = 1; i < m; i++)
16         num[i] += num[i - 1];
17     for(i = n - 1; i >= 0; i--)
18         sa[--num[r[i]]] = i;
19     for(j = 1, p = 1; p < n; j <<= 1, m = p) {
20         for(p = 0, i = n - j; i < n; i++)
21             y[p++] = i;
22         for(i = 0; i < n; i++)
23             if(sa[i] >= j)
24                 y[p++] = sa[i] - j;
25         for(i = 0; i < n; i++)
26             wv[i] = x[y[i]];
27         for(i = 0; i < m; i++) 
28             num[i] = 0;
29         for(i = 0; i < n; i++)
30             num[wv[i]]++;
31         for(i = 1; i < m; i++)
32             num[i] += num[i - 1];
33         for(i = n - 1; i >= 0; i--)
34             sa[--num[wv[i]]] = y[i];
35         swap(x, y);
36         for(p = 1, x[sa[0]] = 0, i = 1; i < n; i++)
37             x[sa[i]] = cmp(y, sa[i - 1], sa[i], j) ? p - 1 : p++;
38     }
39         for(int i=0;i<n;i++) rk[sa[i]]=i;   
40 }
41 void get_height(char *r,int n){
42     int k=0;     
43     for(int i=0;i<n;i++) {
44         if(k) k--;  
45         int j=sa[rk[i]-1]; 
46         while (r[i+k]==r[j+k]) k++;  
47         h[rk[i]]=k;  
48     }  
49 }
50         
51 int main() {
52     gets(s);
53     int len = strlen(s);
54     build_sa(s, sa, len + 1, 130); 
55         for(int i=1;i<=len;i++) 
56             printf("%d ",sa[i]+1); 
57         printf("\n");  
58         get_height(s,len);  
59         for(int i=2;i<=len;i++)  
60             printf("%d ",h[i]); 
61         return 0; 
62 }
View Code

 

posted @ 2019-05-30 12:31  zjxxcn  阅读(135)  评论(0编辑  收藏  举报