HDU - 3336 （Next数组+dp）

本文主要是介绍HDU - 3336 （Next数组+dp），希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

Count the string   HDU - 3336    

题意：给定字符串s，求s的所有前缀作为子串在s中出现的次数和。

思路：考虑dp

dp[i] 表示以第i个字母结尾的所有后缀与s的前缀的匹配次数。

按照最暴力的想法来做的话，肯定我得枚举以第i个字母结尾的每个后缀。

其实也可以这样想：dp[i] 等于s(0,i)最长后缀对结果的贡献加1。然后又kmp里Next数组可以得知，s(0,i)的最长后缀等于Next[i]对应的s的前缀。

由此可得：dp[i] = dp[Next[i]]+1;  dp[0] = 0;

吐槽：此题数据太水。。。

#include <bits/stdc++.h>
using namespace std;
const int mod = 10007;
char s[2000005];
int dp[2000005],Next[2000005];
void getNext(char s[])
{int len = strlen(s),i = 0;int k = -1;Next[0] = -1;while(i < len){if(k == -1 || s[i] == s[k]){Next[++i] = ++k;}else k = Next[k];}
}
int main()
{int t;cin >> t;while(t--){int n;scanf("%d",&n);scanf("%s",s);getNext(s);dp[0] = 0;int ans = 0;for(int i = 1; i <= n; i++){dp[i] = dp[Next[i]]+1;dp[i] %= mod;ans += dp[i];ans %= mod;}printf("%d\n",ans);}return 0;
}

 

这篇关于HDU - 3336 （Next数组+dp）的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---