【hihocoder1445】后缀自动机二·重复旋律5 后缀自动机模板

最新推荐文章于 2020-09-30 12:26:08 发布

愤怒的愣头青

最新推荐文章于 2020-09-30 12:26:08 发布

阅读量432

点赞数

分类专栏：后缀自动机学习资料

本文链接：https://blog.youkuaiyun.com/w_yqts/article/details/79143400

版权

学习资料同时被 2 个专栏收录

9 篇文章

订阅专栏

后缀自动机

5 篇文章

订阅专栏

本文介绍了一种利用AC自动机解决寻找不同子串数量的方法。通过构造AC自动机，文章详细解释了如何计算一个给定字符串中的所有不同子串的数量。此算法适用于音乐旋律等序列数据的不同片段计数。

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

描述

小Hi平时的一大兴趣爱好就是演奏钢琴。我们知道一个音乐旋律被表示为一段数构成的数列。

现在小Hi想知道一部作品中出现了多少不同的旋律？

解题方法提示

输入

共一行，包含一个由小写字母构成的字符串。字符串长度不超过 1000000。

输出

一行一个整数，表示答案。

样例输入
aab
样例输出
5

题解
每个节点的maxlen[i]-maxlen[suf[i]]之和即为答案

代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
char s[1000005];
int cnt,last,len,S,a[1000005];
int l[2000005],trans[2000005][27];
int fa[2000005];ll ans;
inline void add(int x)
{
    int c=a[x],p=last,np=++cnt;last=np;l[np]=x;
    for (;p&&!trans[p][c];p=fa[p]) trans[p][c]=np;
    if (!p) fa[np]=S;
    else
    {
        int q=trans[p][c];
        if (l[p]+1==l[q]) fa[np]=q;
        else
        {
            int nq=++cnt;l[nq]=l[p]+1;
            memcpy(trans[nq],trans[q],sizeof trans[q]);
            fa[nq]=fa[q];fa[np]=fa[q]=nq;
            for (;trans[p][c]==q;p=fa[p]) trans[p][c]=nq;
        }
    }
}
int main()
{
    scanf("%s",s+1);
    last=S=++cnt;
    int len=strlen(s+1);
    for (int i=1;i<=len;i++) a[i]=s[i]-'a';
    for (int i=1;i<=len;i++) add(i);
    for (int i=2;i<=cnt;i++) ans+=l[i]-l[fa[i]];
    printf("%lld",ans);
    return 0;
}