Count the string HDU - 3336（KMP/next数组）

最新推荐文章于 2022-10-26 15:00:00 发布

kelianlee

最新推荐文章于 2022-10-26 15:00:00 发布

阅读量156

点赞数

CC 4.0 BY-SA版权

分类专栏： # 字符串/KMP

本文链接：https://blog.youkuaiyun.com/leekerian/article/details/90649467

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

12 篇文章

订阅专栏

博客围绕给出的串A，探讨求其所有前缀子串在A中出现次数总和的问题。解题思路是利用next数组表示最大前后缀长度，求出所有next数组，用map存储对应长度的next前缀长度，最后加上串的长度n。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

给出一个串A，求A串的所有前缀子串在A中出现的次数求和

思路

next数组表示的最大前后缀的长度，我们只要求出所有的next数组，对于next[i]=k说明前缀长度k的前缀子串出现一次，所以我们开一个map把每个对应长度的next前缀长度存起来，最后加上n就可以了

#include <iostream>
#include <map>

using namespace std;
const int MAXN=2e5+10;
char str[MAXN];
int snext[MAXN],n;
int mod=10007;
void getnext()
{
    int j=0;
    int k=snext[0]=-1;
    while(j<n)
    {
        if(k==-1||str[j]==str[k])
        {
            j++;k++;
            snext[j]=k;
        }
        else k=snext[k];
    }
}


map<int,int> mp;
int main()
{
    int T;cin>>T;
    while(T--)
    {
        mp.clear();
        cin>>n;
        scanf("%s",str);
        getnext();
        for(int i=0;i<=n;i++)
            mp[snext[i]]++;
        int ans=0;
        for(int i=1;i<=n;i++)
            ans=(ans%mod+mp[i]%mod)%mod;
        ans=(ans%mod+n%mod)%mod;
        printf("%d\n",ans);
    }
    return 0;
}