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 | 1
using 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 ;
}