hdu3336-Count the string

Count the string

使用get_next函数多了,渐渐对get_next函数的理解又深了一点点,

这其中dp[i]从1加到n

dp[i]=dp[next[i]]+1;


// File Name: hdu3336.cpp
// Author: bo_jwolf
// Created Time: 2013年04月30日 星期二 20:39:15

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include

using namespace std;
const int maxn = 200005 ;
int dp[ maxn ] ;
const int MOD = 10007 ;
char str[ maxn ] ;
int next[ maxn ];
int len ;
void get_next()
{
	int j = 0 , k = -1 ;
	next[ 0 ] = -1 ;
	len = strlen( str ) ;
	while( j < len )
	{
		if( k == -1 || str[ j ] == str[ k ] )
		{
			j++ ;
			k++ ;
			next[ j ] = k ; 
		}
		else
			k = next[ k ] ;
	}
}

int main()
{
	int n ;
	int Case ;
	int ans ;
	scanf( "%d" ,&Case );
	while( Case-- )
	{
		scanf( "%d" , &n ) ;
		scanf( "%s" , str );
		get_next();
		dp[ 0 ] = 0 ;
		ans = 0 ;
		for(int i = 1 ; i <= n ; i++ )
		{
			dp[ i ] = dp[ next[ i ] ] + 1 ;
			dp[ i ] = dp[ i ] % MOD ;
			ans += dp[ i ];
			ans = ans % MOD ;
		}
		printf( "%d\n" , ans ) ;
	}
return 0;
}






你可能感兴趣的:(一点一点学算法,学习之路,Kmp)