码迷,mamicode.com
首页 > 其他好文 > 详细

[SDOI2016] 生成魔咒 - 后缀自动机,STL

时间:2020-06-22 23:19:14      阅读:60      评论:0      收藏:0      [点我收藏+]

标签:tomat   new   clu   map   one   后缀自动机   inline   ast   个数   

Description

按顺序在一个序列的末尾插入数字,求每次插入后的本质不同子串个数

Solution

后缀自动机增量构造时,已经有的结点的 len 是不会再变化的,而 nq 结点不会产生贡献,于是每次加上 maxlen[newnode]-maxlen[link[newnode]] 即可

由于插入的是数字,所以需要用 map 存储

#include <bits/stdc++.h>
using namespace std;
const int Maxn = 200005;
struct Suffix_Automata {
    int maxlen[Maxn], link[Maxn], Size, Last;
    map<int,int> trans[Maxn];
    int t[Maxn], a[Maxn], cnt[Maxn], f[Maxn];
    long long ans;
    Suffix_Automata() { Size = Last = 1; }
    inline void Extend(int id) {
        int cur = (++ Size), p;
        maxlen[cur] = maxlen[Last] + 1;
        cnt[cur] = 1;
        for (p = Last; p && !trans[p][id]; p = link[p]) trans[p][id] = cur;
        if (!p) link[cur] = 1;
        else {
            int q = trans[p][id];
            if (maxlen[q] == maxlen[p] + 1) link[cur] = q;
            else {
                int clone = (++ Size);
                maxlen[clone] = maxlen[p] + 1;
                trans[clone] = trans[q];
                link[clone] = link[q];
                for (; p && trans[p][id] == q; p = link[p]) trans[p][id] = clone;
                link[cur] = link[q] = clone;
            }
        }
        Last = cur;
        ans += maxlen[cur] - maxlen[link[cur]];
    }
} sam;

int main() {
    ios::sync_with_stdio(false);
    int n;
    cin>>n;
    for(int i=0;i<n;i++) {
        int x;
        cin>>x;
        sam.Extend(x), cout<<sam.ans<<endl;
    }
}


[SDOI2016] 生成魔咒 - 后缀自动机,STL

标签:tomat   new   clu   map   one   后缀自动机   inline   ast   个数   

原文地址:https://www.cnblogs.com/mollnn/p/13179313.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!