BZOJ 4516 [Sdoi2016]生成魔咒

最新推荐文章于 2022-04-05 17:04:07 发布

Endless_Way

最新推荐文章于 2022-04-05 17:04:07 发布

阅读量369

点赞数

CC 4.0 BY-SA版权

分类专栏：字符串-后缀文章标签： BZOJ-4516

本文链接：https://blog.youkuaiyun.com/ziqian2000/article/details/54773273

字符串-后缀专栏收录该内容

18 篇文章

订阅专栏

本文介绍了一种使用后缀自动机处理字符串问题的方法。通过插入字符并更新失败指针来构建自动机，实现了字符串匹配的功能。代码示例展示了如何初始化自动机、插入新字符并维护相关状态。

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

后缀自动机

插入字符更新fail树即可。

#include<cstdio>
#include<map>
#define N 100005
#define A 28
using namespace std;
namespace runzhe2000
{
    typedef long long ll;
    int n; ll ans;
    struct SAM
    {
        map<int,SAM*> next;
        SAM *fail;
        int len;
    } mem[N*2], *tot, *null, *last, *root;
    SAM* newnode(int len)
    {
        SAM *p = ++tot;
        *p = *null; p->len = len;
        return p;
    }
    void init()
    {
        null = tot = mem;
        null->fail = null; null->len = 0;
        last = root = newnode(0);       
    }
    void ins(int v)
    {
        SAM *np = newnode(last->len + 1), *p = last; last = np;
        for(; p != null && !p->next[v]; p = p->fail) p->next[v] = np;
        if(p == null) np->fail = root, ans += np->len;
        else
        {
            SAM *q = p->next[v];
            if(q->len == p->len + 1) np->fail = q, ans += np->len - q->len;
            else
            {
                SAM *nq = newnode(0); *nq = *q; nq->len = p->len + 1;
                np->fail = q->fail = nq; ans += np->len - nq->len;
                for(; p!=null && p->next[v] == q; p = p -> fail) p->next[v] = nq;
            }
        }
    }
    void main()
    {
        scanf("%d",&n);
        init();
        for(int i = 1, v; i <= n; i++)
        {
            scanf("%d",&v);
            ins(v); printf("%lld\n",ans);
        }
    }
}
int main()
{
    runzhe2000::main();
}