后缀数组(O(n)三分实现)

比起WC关于后缀数组的倍增法,要更有效率,由于模版使用的是指针,所以在new和delete处效率会有所消耗

 

来自NIT的后缀数组模版 注意在字符串间加入特殊符号进行区分 避免LCP越界

 

SAMLPE测试为HDU 1403~~

 

 

#include < iostream >
#include < cmath >
#include < cstring >
using   namespace  std;
const   int  maxn = 210000 ;

char  s[maxn];
int  len,k;
int  sa[maxn],rank[maxn],lcp[maxn];
int  num[maxn];

inline  bool  leq( int  a1,  int  a2,  int  b1,  int  b2)
{
     return  (a1  <  b1  ||  a1  ==  b1  &&  a2  <=  b2);
}

inline  bool  leq( int  a1,  int  a2,  int  a3,  int  b1,  int  b2,  int  b3)
{
     return (a1  <  b1  ||  a1  ==  b1  &&  leq(a2, a3, b2, b3));
}

static   void  radixPass( int *  a,  int *  b,  int *  r,  int  n,  int  K)
{
     int *  c  =   new   int [K  +   1 ];
     int  i,sum;
     for (i  =   0 ; i  <=  K; i ++ ) 
        c[i]  =   0 ;
     for (i  =   0 ; i  <  n; i ++ ) 
        c[r[a[i]]] ++ ;
     for (i  =   0 , sum  =   0 ; i  <=  K; i ++ )
    {
         int  t  =  c[i]; 
        c[i]  =  sum; 
        sum  +=  t;
    }
     for (i  =   0 ; i  <  n; i ++ ) 
        b[c[r[a[i]]] ++ ]  =  a[i];
    delete [] c;
}

void  suffixArray( int *  T,  int *  SA,  int  n,  int  K)
{
     int  n0  =  (n  +   2 )  /   3 , n1  =  (n  +   1 )  /   3 , n2  =  n  /   3 , n02  =  n0  +  n2;
     int *  R  =   new   int [n02  +   3 ]; R[n02]  =  R[n02 + 1 ]  =  R[n02  +   2 ]  =   0 ;
     int *  SA12  =   new   int [n02  +   3 ]; SA12[n02]  =  SA12[n02  +   1 ]  =  SA12[n02  +   2 ]  =   0 ;
     int *  R0  =   new   int [n0];
     int *  SA0  =   new   int [n0];
     int  i,j;
     for (i  =   0 , j  =   0 ; i  <  n  +  (n0  -  n1); i ++ )  if (i  %   3   !=   0 ) R[j ++ ]  =  i;
    radixPass(R , SA12, T  +   2 , n02, K);
    radixPass(SA12, R , T  +   1 , n02, K);
    radixPass(R , SA12, T , n02, K);
     int  name  =   0 , c0  =   - 1 , c1  =   - 1 , c2  =   - 1 ;
     for ( i  =   0 ; i  <  n02; i ++ )
    {
         if (T[SA12[i]]  !=  c0  ||  T[SA12[i]  +   1 ]  !=  c1  ||  T[SA12[i]  +   2 ]  !=  c2)
        {
            name ++ ; c0  =  T[SA12[i]]; c1  =  T[SA12[i]  +   1 ]; c2  =  T[SA12[i]  +   2 ];
        }
         if (SA12[i]  %   3   ==   1 ) { R[SA12[i]  /   3 ]  =  name; }
         else { R[SA12[i]  /   3   +  n0]  =  name; }
    }
     if (name  <  n02)
    {
        suffixArray(R, SA12, n02, name);
         for ( int  i  =   0 ; i  <  n02; i ++ ) R[SA12[i]]  =  i  +   1 ;
    }
     else
         for ( int  i  =   0 ; i  <  n02; i ++ ) SA12[R[i]  -   1 ]  =  i;
     for (i  =   0 , j  =   0 ; i  <  n02; i ++ )  if (SA12[i]  <  n0) R0[j ++ ]  =   3   *  SA12[i];
    radixPass(R0, SA0, T, n0, K);
     for ( int  p  =   0 , t  =  n0  -  n1, k  =   0 ; k  <  n; k ++ )
    {
         #define  GetI() (SA12[t] < n0 ? SA12[t] * 3 + 1 : (SA12[t] - n0) * 3 + 2)
         int  i  =  GetI();
         int  j  =  SA0[p];
         if (SA12[t]  <  n0  ?
        leq(T[i], R[SA12[t]  +  n0], T[j], R[j  /   3 ]) : leq(T[i],T[i  +   1 ],R[SA12[t]  -  n0  +   1 ], T[j],T[j  +   1 ],R[j  /   3   +  n0]))
        {
            SA[k]  =  i; t ++ ;
             if (t  ==  n02)
                 for (k ++ ; p  <  n0; p ++ , k ++ ) SA[k]  =  SA0[p];
        }
         else {
            SA[k]  =  j;
             if ( ++ p  ==  n0) for (k ++ ; t  <  n02; t ++ , k ++ ) SA[k]  =  GetI();
        }
    }
    delete [] R; delete [] SA12; delete [] SA0; delete [] R0;
}

void  LCP( char   * str) 
{ 
     int  i,j,k; 
     for (k = i = 0 ;i < len;i ++ ) 
    { 
         if ( rank[i]  ==  len  -   1  ) lcp[ rank[i] ]  =  k  =   0 ; 
         else  
        { 
             if ( k  >   0  )    k -- ; 
            j  =  sa[ rank[i]  +   1 ]; 
             for (;str[i + k]  ==  str[j + k];k ++ ); 
            lcp[rank[i]]  =  k; 
        } 
    } 
}

int  main(){

     int  i, n;

    scanf( " %s " ,s);
    n = strlen(s);
    s[n] = ' A ' - 1 ;
    scanf( " %s " ,s + n + 1 );
    len = strlen(s);
     for (i = 0 ;i < len;i ++ )
        num[i] = s[i] - ' A ' + 1 ;
    suffixArray(num, sa, len,  60 );
     for (i = 0 ;i < len;i ++ )
        rank[sa[i]] = i;

    LCP(s);

     int  ans = 0 ;
     for (i = 1 ;i < len;i ++ )
         if (sa[i] < n  &&  sa[i - 1 ] > n   ||   sa[i] > n  &&  sa[i - 1 ] < n)
             if (lcp[i - 1 ] > ans) ans = lcp[i - 1 ];

    printf( " %d\n " ,ans);

     return   0 ;
}