[HDU 3336] Count the string

题目传送-HDU3336

题意：

给你一个长度为\(n\)的字符串\(S\),问这个字符串的所有前缀在字符串中出现的次数之和(mod \(10007\))
\(n\le 200000\)

题解：

1.考虑KMP，并记录\(f_i\)表示以\(i\)为结尾的所有串中是前缀的个数,那么答案显然是所有\(f\)之和
根据KMP中nxt数组的定义,\(f_i=f_{nxt_i}+1\)(如果这里不懂的自行搜索KMP理解)
2.不用动脑子的后缀自动机.

过程：

顺利AC

代码：

const int LEN=200010;
int n,nxt[LEN];
char s[LEN];
int f[LEN];
signed main() {
	int T; read(T);
	while(T--) {
		read(n); scanf("%s",s+1);
		int j=0;
		for(int i=2;i<=n;i++) {
			while(s[j+1]!=s[i] && j) j=nxt[j];
			if(s[j+1]==s[i]) ++j;
			nxt[i]=j;
		}
		for(int i=1;i<=n;i++)
			f[i]=f[nxt[i]]+1;
		ll ans=0;
		for(int i=1;i<=n;i++)
			(ans+=1ll*f[i])%=10007;
		printf("%lld\n",ans);
	}

	return 0;
}
/*
2
4
abab
4
abab
*/

用时：15min