hdoj3336(kmp算法next数组的应用)

梦想的初衷 提交于 2019-12-03 10:06:29

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

题意:给定长为n(<=2e5)的字符串s,求s的每个前缀在s中出现的次数之和。

思路:

  用dp[i]表示以s[i]为结尾的子串是s的某一种前缀的方案数,那么dp[i]=dp[nex[i]]+1,因为[nex[i]-(i-nex[i])+1 , nex[i]] 和 [nex[i]+1 , i] 是一样的,所以计算后者时只用把前者+1即可(+1是因为以[1,i]为前缀的个数加一)。

AC代码:

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;

const int maxn=2e5+5;
const int MOD=1e4+7;
int T,n,ans,nex[maxn],dp[maxn];
char s[maxn];

void get_next(){
    int j;
    j=nex[0]=-1;
    for(int i=1;i<n;++i){
        while(j>-1&&s[i]!=s[j+1]) j=nex[j];
        if(s[i]==s[j+1]) ++j;
        nex[i]=j;
    }
}

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

 

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!