HDU - 3336 Count the string kmp+dp

最新推荐文章于 2020-07-21 21:21:14 发布

mmk27

最新推荐文章于 2020-07-21 21:21:14 发布

阅读量155

点赞数

分类专栏： dp Android

本文链接：https://blog.csdn.net/mmk27_word/article/details/98608067

版权

dp 同时被 2 个专栏收录

39 篇文章 0 订阅

订阅专栏

Android

11 篇文章 0 订阅

订阅专栏

题目链接：https://cn.vjudge.net/problem/HDU-3336

题意：字符串所有前缀的出现次数

题解：dp[ i ] 表示以i为结尾 i 前缀的个数，所以dp[i] = dp[nex[i] ] + 1

#include <bits/stdc++.h>
using namespace std;
const int N = 200100;
const int mod = 10007;
int nex[N], dp[N];
void get_nex(char *s) {
	int len = strlen(s);
	int i = 0, j = -1;
	nex[0] = -1;
	while(i < len) {
		if(j == -1 || s[i] == s[j])
			nex[++i] = ++j;
		else
			j = nex[j];
	}
}
int main() {
	int T, n;
	char s[N];
	long long ans;
	scanf("%d", &T);
	while(T--) {
		scanf("%d %s", &n, s);
		get_nex(s);
		dp[0] = 0;
		ans = 0;
		for(int i = 1; i <= n; i++) {
			dp[i] = dp[nex[i]] + 1;
			ans = (ans + dp[i]) % mod;
		}
		printf("%d\n", ans);
	}
	return 0;
}