2019杭电多校Day5 1006

cloudy_happy

于 2019-08-05 20:41:39 发布

阅读量130

点赞数

分类专栏： KMP

本文链接：https://blog.youkuaiyun.com/cloudy_happy/article/details/98517133

版权

KMP 专栏收录该内容

7 篇文章

订阅专栏

本文详细介绍了一种基于KMP算法的扩展应用，通过预处理生成nxt数组，实现字符串匹配过程中对全部匹配前缀的有效计数。文章提供了完整的代码实现，展示了如何利用此方法在字符串搜索中进行优化。

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

思路：

扩展kmp，预处理出nxt数组相加，对于全部匹配的前缀减1就行了

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 1e6 + 10;
ll nxt[maxn],ans;
void getnext(string t)
{
    memset(nxt,0,sizeof(nxt));
    ll len=t.length();
    nxt[0]=len;
    ll a,p;
    a=1;
    while( a<len && t[a]==t[a-1]) a++; // 求出长度为1的时候 解为多少
    nxt[1]=a-1;
    a=1;
    for(ll i=2;i<len;i++) // 后续的按照算法来就好
    {
        p=a+nxt[a]-1;
        if((i-1)+nxt[i-a] < p ) nxt[i]=nxt[i-a];// 第一种情况 没有超过等于的部分
        else // 超过的话就不好直接用nxt的定义 需要后续的遍历
        {
            ll j = (p - i + 1) > 0 ? (p - i + 1) : 0;
            while(i + j < len && t[i+j] == t[j]) j++;
            nxt[i]=j;
            a=i;
        }
    }
}
int main()
{
    ios::sync_with_stdio(false);
    int t;
    cin>>t;
    while (t --)
    {
        string s;
        cin>>s;
        ans = 0;
        getnext(s);
        for (int i = 0;i < s.size();i ++)
        {
            ans += nxt[i];
            if (nxt[i]+i==s.size()) ans--;
        }
        cout<<ans<<endl;
    }
    return 0;
}