HDU 3336 Count the string(KMP:串前缀匹配自身+DP)

HDU 3336 Count the string(KMP:串前缀匹配自身+DP)

http://acm.hdu.edu.cn/showproblem.php?pid=3336

题意:

        给你一个串,求用该串所有前缀去 匹配本身这个串的次数 的总和。比如串abab，它的前缀有a，ab，aba，abab。那么拿这4个前缀去匹配abab自身分别有2,2,1,1个匹配点，所以总和为2+2+1+1=6。

分析:

        首先想一想,每个前缀是不是都要去和 串的以某个节点结尾的后缀去匹配。

        其实我们想想比如对于位置i来说，现在我们求出了f[i]的值，那么[0，f[i]-1]串就是一个能匹配i-1位置的最长的前缀串。且我们根据f[i]的值可以定位其他所有可匹配的前缀。KMP的思想就是找出每个位置i的可匹配最大前缀j。

        令dp[i]表示S[0,i-1]串的后缀能匹配的前缀个数，那么dp[i]=dp[next[i]]+1, dp[1]=1正好表示串s[0]的后缀只能匹配串s[0].

AC代码:

#include 
#include
#include
#include
#include
using namespace std;
const int MAXM=200000+100;
const int MOD=10007;
char P[MAXM];
int f[MAXM],dp[MAXM];
int m;
void getFail(char *P,int *f)
{
    f[0]=f[1]=0;
    for(int i=1;i