后缀数组

Posted 2021-01-02 jaydenouyang

https://blog.csdn.net/qq_35640373/article/details/70168683

 1 //hdu 3518
 2 #include <iostream>
 3 #include <string.h>
 4 #include <stdlib.h>
 5 #include <cmath>
 6 #define LL long long
 7 using namespace std;
 8 
 9 #define maxn 1020
10 int max(int x,int y){ return x>y?x:y;}
11 int min(int x,int y){ return x<y?x:y;}
12 int wa[maxn],wb[maxn],wv[maxn],WS[maxn];
13 
14 int cmp(int *r,int a,int b,int l){
15     return r[a]==r[b]&&r[a+l]==r[b+l];
16 }
17 
18 void da(int *r,int *sa,int n,int m){
19     int i,j,p,*x=wa,*y=wb,*t;
20     for(i=0;i<m;i++) WS[i]=0;
21     for(i=0;i<n;i++) WS[x[i]=r[i]]++;
22     for(i=1;i<m;i++) WS[i]+=WS[i-1];
23     for(i=n-1;i>=0;i--) sa[--WS[x[i]]]=i;
24     for(j=1,p=1;p<n;j*=2,m=p){
25         for(p=0,i=n-j;i<n;i++) y[p++]=i;
26         for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
27         for(i=0;i<n;i++) wv[i]=x[y[i]];
28         for(i=0;i<m;i++) WS[i]=0;
29         for(i=0;i<n;i++) WS[wv[i]]++;
30         for(i=1;i<m;i++) WS[i]+=WS[i-1];
31         for(i=n-1;i>=0;i--) sa[--WS[wv[i]]]=y[i];
32         for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)
33             x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
34     }
35     return;
36 }
37 
38 int Rank[maxn],height[maxn];
39 
40 void calheight(int *r,int *sa,int n){
41     int i,j,k=0;
42     for(i=1;i<=n;i++) Rank[sa[i]]=i;
43     for(i=0;i<n;height[Rank[i++]]=k)
44         for(k?k--:0,j=sa[Rank[i]-1];r[i+k]==r[j+k];k++);
45     return;
46 }
47 
48 int r[maxn],sa[maxn];
49 char s[maxn];
50 
51 int ok(int k,int n){
52     int i,j,ret=0;
53     int maxx,minx;
54     maxx=-1;
55     minx=9999999;
56     for(i=1;i<=n;i++){
57         if(height[i]>=k)//分组满足大于k的
58         {
59             maxx=max(maxx,sa[i]);
60             minx=min(minx,sa[i]);
61         }
62         else
63         {
64             if(maxx-minx>=k)//之前满足条件就要+1；
65                 ret++;
66             maxx=sa[i];
67             minx=sa[i];
68         }
69     }
70     if(maxx-minx>=k)
71         ret++;
72     return ret;
73 }
74 
75 int main(){
76     int i,j;
77     while(cin>>s){
78         if(s[0]==‘#‘) break;
79         int len=strlen(s);
80         for(i=0;i<len;i++)
81             r[i]=s[i];
82         r[len]=0;
83         da(r,sa,len+1,123);
84         calheight(r,sa,len);
85         LL ans=0;
86         for(i=1;i<len;i++){
87             ans+=ok(i,len);
88         }
89         cout<<ans<<endl;
90     }
91     return 0;
92 }