HDU - 3336 Count the string（NXT数组，DP）

小松萘

于 2020-02-18 11:34:41 发布

阅读量142

点赞数

分类专栏：字符串---KMP

本文链接：https://blog.csdn.net/weixin_44116061/article/details/104371834

版权

字符串---KMP 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

🍮 🍮 🍮

（1）求每个前缀在字符串中出现的次数 ----> 枚举每个位置结尾的字符串中有多少前缀，求和
（2） i 位置能匹配的最长前缀（除1~i这个串以外）为nxt[ i ]，dp[ i ] = dp[nxt[ i ] ] + 1；

char a[MAXN];
int  n,dp[MAXN],nxt[MAXN];
void get_nxt()
{
    nxt[0]=-1;
    int j=-1,i=0;
    while(i<n)
    {
        if(j==-1||a[i]==a[j]) nxt[++i]=++j;
        else j=nxt[j];
    }
}
signed main()
{
    int T;cin>>T;
    while(T--)
    {
        cin>>n>>a;
        get_nxt();
        memset(dp,0,sizeof(dp));
        int ans=0;
        for(int i=1;i<=n;++i)
        {
            dp[i]=dp[nxt[i]]+1;
            ans=(ans+dp[i])%10007;
        }
        cout<<ans<<endl;
    }
    return 0;
}