hdu 4763 Theme Section (扩展kmp)

来源:互联网 发布:c语言好的书籍 编辑:程序博客网 时间:2024/05/16 12:49

hdu 4763 Theme Section (扩展kmp)

题目大意:给出一个字符串,问这个字符串的最长Theme Section是多长,Theme Section要符合以下条件,必须出现在字符串的开头,结尾,和中间,而且不能有重叠。

很抱歉,之前写了个很挫o(nlogn)的解法,误导了大家,后来我仔细想了想,其实用扩展kmp可以o(n)的实现。下面给出具体的实现方法:

解题思路:还是先扩展kmp,把next数组处理出来,next[i]表示以i为开头的后缀与该字符串的最长公共前缀的长度。对于以长度为i的后缀,若它要能成为答案,那么就要满足在[i,len-2*i]这个区间里找出一个后缀,这个后缀与字符串的最长公共前缀要大于等于i。在我之前写的博文里面是用了个线段树维护这个最大值,但其实,仔细想想,如果我们把i的长度从大往小的枚举,那么我们每减小一个长度,区间的左端点就会减1,右端点就会加2,而且区间是一直增大的,那么我们就可以在每次询问后,在更新下这个值就好了,整体复杂度只有o(n)了。

#include<stdio.h>#include<string.h>#include<algorithm>#define lson l , m , rt << 1#define rson m + 1 , r , rt << 1 | 1using namespace std ;const int maxn = 1111111 ;int max ( int a , int b ) { return a > b ? a : b ; }int p[maxn] ;void get_p (const char *T){     int len=strlen(T),a=0;     int i , k ;     p[0]=len;     while(a<len-1 && T[a]==T[a+1]) a++;     p[1]=a;     a=1;     for( k=2;k<len;k++){         int fuck=a+p[a]-1,L=p[k-a];         if( (k-1)+L >= fuck){             int j = (fuck-k+1)>0 ? (fuck-k+1) : 0;             while(k+j<len && T[k+j]==T[j]) j++;             p[k]=j;             a=k;         }         else p[k]=L;     }}char s[maxn] ;int main () {    int t , i ;    scanf ( "%d" , &t ) ;    while ( t -- ) {        scanf ( "%s" , s ) ;        get_p ( s ) ;        int n = strlen ( s ) ;        int mx = 0 ;        int ans = 0 ;        for ( i = n / 3 ; i <= n - 2 * ( n / 3 ) ; i ++ ) mx = max ( mx , p[i] ) ;        int l = n / 3 , r = n - 2 * ( n / 3 ) ;    //    printf ( "l = %d , r = %d\n" , l , r ) ;        for ( i = n / 3 ; i >= 1 ; i -- ) {      //      printf ( "p[%d] = %d , mx = %d\n" , n - i , p[n-i] , mx ) ;            if ( p[n-i] == i && mx >= i ) {                ans = i ;                break ;            }     //       printf ( "l = %d , r = %d " , l , r ) ;            mx = max ( mx , max ( p[--l] , p[++r] ) ) ;            mx = max ( mx , p[++r] ) ; //           printf ( "mx = %d\n" , mx ) ;        }        printf ( "%d\n" , ans ) ;    }    return 0 ;}