[JZOJ 5178] So many prefix? {KMP+DP}

最新推荐文章于 2024-01-16 12:32:17 发布

原创最新推荐文章于 2024-01-16 12:32:17 发布 · 341 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#KMP #DP

动态规划（/线性DP）同时被 2 个专栏收录

87 篇文章

订阅专栏

KMP算法

9 篇文章

订阅专栏

本文深入探讨了KMP+DP算法在字符串匹配问题中的应用，通过实例详细讲解了如何利用KMP算法的next数组与DP思想求解特定字符串中偶数子串的出现次数，提供了完整的解题思路及C++代码实现。

题目

题目链接：

Sample Input
样例一：
abababc

样例二：
isdashagayisdashagaydashisnotagaydashisnotagay
Sample Output
样例一：
6
样例二：
30

解题思路`KMP+DP`

$K M P + D P$ ，考虑 $K M P$ 中的 $n e x t [i]$ ，代表最大的 $k (k! = i)$ 使 $‘ s [1] s [2] \dots s [k] = = s [i - k + 1] s [i - k] \dots s [i] ’$ ，那么我们设 $f [i]$ 代表以 $i$ 前缀 $‘ s [1] s [2] \dots s [i] ’$ 内所有偶数子串出现的次数（包含本身），得到：
$2==1f_i=\left\{\begin{matrix} 1+f[next[i]]\ |\ i\ \%\ 2==0 \\ f[next[i]]\ |\ i\ \ \%\ 2==1 \end{matrix}\right.$

代码

#include<cstdio>
#include<cstring>
#define rr register
using namespace std; 
char s[200010]; 
int len,k,next[200010],f[200010],ans; 
int main()
{
	scanf("%s",s+1); len=strlen(s+1); 
	for(rr int i=2;i<=len;i++)
	 {
	 	while (k&&s[k+1]!=s[i]) k=next[k]; 
	 	if (s[k+1]==s[i]) next[i]=++k; 
	 }
	for (rr int i=2;i<=len;i++)
	 {
		if (i%2==0) f[i]++; 
		f[i]+=f[next[i]]; 
		ans+=f[i]; 
	 }
	printf("%d",ans); 
}