hust 1328 String (kmp+dp)

题意：

给出一个字符串，问这个字符串前缀出现的次数。

题解：

之前hdu的一道题目一样，但是这次意识到上次的贪心算法是错的，ababa这个数据就可以否定了，不过不知道为什么竟然ac了。这题先算出next数组，然后用dp, dp[i]=dp[next[i]]+1，同时统计ans+=dp[i];这是一种dp的解法，还有一种方法是倒着推 dp[next[i]]+=dp[i]，dp初始化为1。

#include<iostream>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
#include<map>
#include<set>
using namespace std;
#define B(x) (1<<(x))
typedef long long ll;
const int oo=0x3f3f3f3f;
const ll OO=1LL<<61;
const ll MOD=10007;
const int maxn=100000+5;
char str[maxn];
int next[maxn],dp[maxn];

void get_next(char s[],int len)
{
    int i=0;
    next[i]=-1;
    int j=-1;
    while(i<len)
    {
        if(j==-1||s[i]==s[j])
        {
            i++;j++;
            next[i]=j;
        }
        else j=next[j];
    }
}

int main()
{
    int len,T;
    ll ans;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%s",str);
        len=strlen(str);
        get_next(str,len);
        memset(dp,0,sizeof dp);
        ans=0;
        for(int i=1;i<=len;i++)
        {
            dp[i]=dp[next[i]]+1;
            ans+=dp[i];
        }
        cout<<ans<<endl;
    }
    return 0;
}

#include<iostream>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
#include<map>
#include<set>
using namespace std;
#define B(x) (1<<(x))
typedef long long ll;
const int oo=0x3f3f3f3f;
const ll OO=1LL<<61;
const ll MOD=10007;
const int maxn=100000+5;
char str[maxn];
int next[maxn];
int dp[maxn],mark[maxn];

void get_next(char s[],int len)
{
    int i=0;
    next[i]=-1;
    int j=-1;
    while(i<len)
    {
        if(j==-1||s[i]==s[j])
        {
            i++;j++;
            next[i]=j;
        }
        else j=next[j];
    }
}

int main()
{
    int len,ans,T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%s",str);
        len=strlen(str);
        get_next(str,len);
        ans=0;
        for(int i=0;i<=len;i++)
            dp[i]=1;
        for(int i=len;i>=1;i--)
            dp[next[i]]+=dp[i];
        for(int i=1;i<=len;i++)
            ans+=dp[i];
        printf("%d\n",ans);;
    }
    return 0;
}