kmp -- Count the string HDU - 3336

Count the string HDU - 3336

题意：
T次询问，每次给你一个数 n 和字符串 s ，n为s的长度。问你s的所有前缀要和s串匹配时要比较多少次。

思路：

• next求出的就是和前缀匹配的最大长度，若长度大于0代表和前缀匹配，那么个数增加1，然后缩小前缀再进行判断。

• 如 ababa 当我们匹配到a的时候，得到的next[5] = 3, 这时候次数增加1， 再匹配一下next[3] = 1,值也大于0那么次数还要增加， 再匹配next[1] = 0, 等于0了代表没有前缀和它相等了。

code：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn = 2e5 + 5;
const int mod = 1e4 + 7;
char s[maxn];
int nxt[maxn];

int get_nxt(int m){
	int i = 0, j = -1;
	nxt[0] = -1;
	int cot = m;
	while(i < m){
		if(j == -1 || s[i] == s[j]){
			 nxt[++i] = ++j;
			 int t = j;
			 while(t){
			 	t = nxt[t];
			 	cot++;
			 }
		}
		else j = nxt[j];
		cot++;
//		printf("i = %d j = %d cot = %d\n", i, j, cot);
	}
	return cot % mod;
}
int main(){
	int T, n;
	scanf("%d", &T);
	while(T--){
		scanf("%d%s", &n, s);
		int len = strlen(s);
		int cot = get_nxt(len);
		printf("%d\n", cot);
		
	}
}