【练习03】 KMP 1006 Count the string

如果枚举所有前缀并且调用find函数会超时，但是我们思考一下就能发现只要遍历数组f，当f的值非0，count就加1(包括f[n])，最后加上字符串长度n就是最终的结果。（仔细理解)

最后还要注意的是题目要求输出mod 10007.

//模板开始
#include <string>   
#include <vector>   
#include <algorithm>   
#include <iostream>   
#include <sstream>   
#include <fstream>   
#include <map>   
#include <set>   
#include <cstdio>   
#include <cmath>   
#include <cstdlib>   
#include <ctime>
#include<iomanip>
#include<string.h>
#define SZ(x) (int(x.size()))
using namespace std;

int toInt(string s){
	istringstream sin(s); 
	int t; 
	sin>>t; 
	return t;
}
template<class T> string toString(T x){
	ostringstream sout; 
	sout<<x; 
	return sout.str();
}
typedef long long int64;
int64 toInt64(string s){
	istringstream sin(s); 
	int64 t; 
	sin>>t;
	return t;
}
template<class T> T gcd(T a, T b){ 
	if(a<0) 
		return gcd(-a, b);
	if(b<0) 
		return gcd(a, -b);
	return (b == 0)? a : gcd(b, a % b);
}
//模板结束（通用部分）

#define ifs cin

void getFail(char* P, int* f)
{
	int m = strlen(P);
	f[0] = 0;
	f[1] = 0;
	for(int i = 1; i < m; i++)
	{
		int j = f[i];
		while(j && P[i] != P[j])
		{
			j = f[j];
		}
		f[i + 1]=P[i]==P[j]?j+1:0;
	}
}

int find(char* T, char*P, int*f)		//KMP改进版本  
{  
	int count = 0;  
	int n = strlen(T), m = strlen(P);  
	getFail(P, f);  
	int j = 0;  
	for(int i = 0; i < n; i++)  
	{  
		while(j && P[j] != T[i])  
		{  
			j = f[j];  
		}  
		if(P[j] == T[i])  
		{  
			j++;  
		}  
		if(j == m)		//这里和常规的KMP不同  
		{  
			count++;  
			j = 0;  
			continue;  
		}  
	}  
	return count;  
}  

//【练习03】 KMP 1006 Count the string
#define MAX_SIZE 200005
char P[MAX_SIZE];
int f[MAX_SIZE];
char T[MAX_SIZE];

int main()
{
	//ifstream ifs("shuju.txt", ios::in);
	int cases;
	int n;
	static int count;
	ifs>>cases;
	for(int i = 0; i < cases; i++)
	{	
		ifs>>n;
		ifs>>P;
		count = n;
		getFail(P, f);
		for(int j = 0; j <= n; j++)
		{
			if(f[j] != 0)
			{
				count++;
			}
		}
		cout<<count % 10007<<endl;
	}
	return 0;
}