【BZOJ4516】【SDOI2016】生成魔咒（SAM）

Description

魔咒串由许多魔咒字符组成，魔咒字符可以用数字表示。例如可以将魔咒字符 1、2 拼凑起来形成一个魔咒串 [1,2]。
一个魔咒串 S 的非空字串被称为魔咒串 S 的生成魔咒。
例如 S=[1,2,1] 时，它的生成魔咒有 [1]、[2]、[1,2]、[2,1]、[1,2,1] 五种。S=[1,1,1] 时，它的生成魔咒有 [1]、[1,1]、[1,1,1] 三种。最初 S 为空串。共进行 n 次操作，每次操作是在 S 的结尾加入一个魔咒字符。每次操作后都需要求出，当前的魔咒串 S 共有多少种生成魔咒。

---

Solution

算是SAM模板题吧。。
每次新加入一个数字（实质是一个前缀）后，贡献为$len_i-len_{fa_i}$($fa_i$为后缀链)

额，输入的数有点大，ch[u][c]用std::map<int,int>或者__gnu_pbds::cc_hash_table都可以。ljBZOJ似乎用不了unordered_map吧。

---

Source

/************************************************
 * Au: Hany01
 * Date: Apr 1st, 2018
 * Prob: [BZOJ4516][SDOI2016] 生成魔咒
 * Email: hany01@foxmail.com
************************************************/

#include<bits/stdc++.h>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/hash_policy.hpp>

using namespace std;
using namespace __gnu_pbds;

typedef long long LL;
typedef pair<int, int> PII;
#define File(a) freopen(a".in", "r", stdin), freopen(a".out", "w", stdout)
#define rep(i, j) for (register int i = 0, i##_end_ = (j); i < i##_end_; ++ i)
#define For(i, j, k) for (register int i = (j), i##_end_ = (k); i <= i##_end_; ++ i)
#define Fordown(i, j, k) for (register int i = (j), i##_end_ = (k); i >= i##_end_; -- i)
#define Set(a, b) memset(a, b, sizeof(a))
#define Cpy(a, b) memcpy(a, b, sizeof(a))
#define x first
#define y second
#define pb(a) push_back(a)
#define mp(a, b) make_pair(a, b)
#define ALL(a) (a).begin(), (a).end()
#define SZ(a) ((int)(a).size())
#define INF (0x3f3f3f3f)
#define INF1 (2139062143)
#define Mod (1000000007)
#define debug(...) fprintf(stderr, __VA_ARGS__)
#define y1 wozenmezhemecaia

template <typename T> inline bool chkmax(T &a, T b) { return a < b ? a = b, 1 : 0; }
template <typename T> inline bool chkmin(T &a, T b) { return b < a ? a = b, 1 : 0; }

inline int read()
{
    register int _, __; register char c_;
    for (_ = 0, __ = 1, c_ = getchar(); c_ < '0' || c_ > '9'; c_ = getchar()) if (c_ == '-') __ = -1;
    for ( ; c_ >= '0' && c_ <= '9'; c_ = getchar()) _ = (_ << 1) + (_ << 3) + (c_ ^ 48);
    return _ * __;
}

const int maxn = 100005;

int fa[maxn << 1], len[maxn << 1], tot = 1, las = 1;
cc_hash_table<int, int> ch[maxn << 1];

inline int extend(int c)
{
    register int np = ++ tot, p = las;
    las = np, len[np] = len[p] + 1;
    while (p && !ch[p][c]) ch[p][c] = np, p = fa[p];
    if (!p) fa[np] = 1;
    else {
        register int q = ch[p][c];
        if (len[q] == len[p] + 1) fa[np] = q;
        else {
            register int nq = ++ tot;
            fa[nq] = fa[q], ch[nq] = ch[q], len[nq] = len[p] + 1;
            fa[np] = fa[q] = nq;
            while (p && ch[p][c] == q) ch[p][c] = nq, p = fa[p];
        }
    }
    return len[np] - len[fa[np]];
}

int main()
{
#ifdef hany01
    File("bzoj4516");
#endif

    static int n = read();
    static LL Ans = 0;

    For(i, 1, n) {
        register int ne = read();
        printf("%lld\n", Ans += extend(ne));
    }

    return 0;
}
//今宵酒醒何处？杨柳岸，晓风残月。
//    -- 柳永《雨霖铃·寒蝉凄切》