URAL 1297 求最长回文字符串

  1 #include <cstdio>
  2 #include <cstring>
  3 #include <iostream>
  4 #include <cmath>
  5 using namespace std;
  6 #define MAXN 2010
  7 int sa[MAXN] , height[MAXN] , _rank[MAXN];
  8 int wa[MAXN] , wb[MAXN] , wsf[MAXN] , wv[MAXN] , r[MAXN];
  9 int dp[MAXN][15];
 10 char str[MAXN];
 11 
 12 int cmp(int *r , int a , int b , int l)
 13 {
 14     return r[a]==r[b] && r[a+l]==r[b+l];
 15 }
 16 
 17 void getSa(int *r , int *sa , int n , int m)
 18 {
 19     int i,j,p;
 20     int *x=wa , *y=wb , *t;
 21 
 22     for(i=0 ; i<m ; i++) wsf[i]=0;
 23     for(i=0 ; i<n ; i++) wsf[x[i]=r[i]]++;
 24     for(i=1 ; i<m ; i++) wsf[i]+=wsf[i-1];
 25     for(i=n-1 ; i>=0 ; i--) sa[--wsf[x[i]]]=i;
 26 
 27     p=1;
 28     for(j=1 ; p<n ; j*=2 , m=p)
 29     {
 30         for(p=0 , i=n-j ; i<n ; i++) y[p++]=i;
 31         for(i=0 ; i<n ; i++) if(sa[i]>=j) y[p++]=sa[i]-j;
 32 
 33         for(i=0 ; i<n ; i++) wv[i]=x[y[i]];
 34         for(i=0 ; i<m ; i++) wsf[i]=0;
 35         for(i=0 ; i<n ; i++) wsf[wv[i]]++;
 36         for(i=1 ; i<m ;  i++) wsf[i]+=wsf[i-1];
 37         for(i=n-1 ; i>=0 ; i--)sa[--wsf[wv[i]]]=y[i];
 38 
 39         t=x , x=y , y=t;
 40         x[sa[0]]=0;
 41         for(i=1,p=1;i<n;i++)
 42             x[sa[i]] = cmp(y , sa[i-1] , sa[i] , j)?p-1:p++;
 43     }
 44     return ;
 45 }
 46 
 47 void callHeight(int *r , int *sa , int n)
 48 {
 49     int i,j,k=0;
 50     for(i=1 ; i<=n ; i++) _rank[sa[i]]=i;
 51   //  for(i=0 ; i<n ; i++) cout<<"_rank: "<<_rank[i]<<endl;
 52     for(i=0 ; i<n ; height[_rank[i++]]=k)
 53         for(k?k--:0 , j=sa[_rank[i]-1]; r[i+k]==r[j+k] ; k++);
 54     return ;
 55 }
 56 
 57 
 58 void ST(int n)
 59 {
 60     //将height数组进行RMQ询问
 61     memset(dp , 0x3f , sizeof(dp));
 62     for(int i=1 ; i<=n ; i++) dp[i][0]=height[i];
 63     for(int j=1 ; (1<<(j-1))<n+1 ; j++)
 64         for(int i=1 ; i<=n ; i++)
 65             dp[i][j] = min(dp[i][j-1] , dp[i+(1<<(j-1))][j-1]);
 66 
 67     return;
 68 }
 69 
 70 int get_min(int s , int t)
 71 {
 72     int len = t-s+1;
 73     int l=floor(log(len*1.0)/log(2.0));
 74     return min(dp[s][l] , dp[t-(1<<l)+1][l]);
 75 }
 76 
 77 void solve(int n , int &left , int &right)
 78 {
 79     int ans = 0;
 80     //奇数情况
 81     for(int i=0 ; i<n ; i++){
 82         int j=2*n-i;
 83         int s=min(_rank[i] , _rank[j]);
 84         int t=max(_rank[i] , _rank[j]);
 85         int tmp = get_min(s+1 , t);
 86         if(ans<tmp*2-1){
 87             ans = tmp*2-1;
 88             left=i-tmp+1 , right=i+tmp-1;
 89         }
 90     }
 91     //偶数情况
 92     for(int i=0 ; i<n ; i++){
 93         int j = 2*n-i+1;
 94         int s=min(_rank[i] , _rank[j]);
 95         int t=max(_rank[i] , _rank[j]);
 96         int tmp = get_min(s+1 , t);
 97         if(ans<tmp*2){
 98             ans = tmp*2;
 99             left=i-tmp , right=i+tmp-1;
100         }
101     }
102     return;
103 }
104 
105 int main()
106 {
107   //  freopen("a.in" , "r" , stdin);
108     while(~scanf("%s" , str))
109     {
110         int len = strlen(str);
111         r[len]=1;
112         for(int i=0 ; i<len ; i++){
113             r[i] = (int)str[i];
114             r[2*len-i] = r[i];
115         }
116         r[2*len+1]=0;
117 
118         getSa(r , sa , 2*len+2 , 255);
119         callHeight(r , sa , 2*len+1);
120         //建立RMQ查询数组
121         ST(2*len+1);
122         int left=1 , right=1;
123         solve(len , left , right);
124         str[right+1]=‘\0‘;
125         printf("%s\n" , str+left);
126     }
127     return 0;
128 }