BZOJ4516: [Sdoi2016]生成魔咒 后缀自动机

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<cmath>
#include<algorithm>
#include<cstdlib>
#include<map>
#define N 200005
#define ll long long
using namespace std;
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;
}
int fa[N],mx[N];
int rt,lst,tot;
map<int,int> son[N];
int n;
ll ans=0;
void ins(int x){
  int p=lst,np=++tot;
  mx[np]=mx[p]+1;
  while(p&&!son[p][x]){
    son[p][x]=np;p=fa[p];
  }
  if(!p)fa[np]=rt;
  else{
    int q=son[p][x];
    if(mx[q]==mx[p]+1)fa[np]=q;
    else{
        int nq=++tot;
        mx[nq]=mx[p]+1;
        son[nq]=son[q];
        fa[nq]=fa[q];
        fa[np]=fa[q]=nq;
        while(son[p][x]==q&&p){
            son[p][x]=nq;p=fa[p];
        }
    }
  }
  lst=np;
  ans+=mx[np]-mx[fa[np]];
  printf("%lld\n",ans);
}
int main(){
    n=read();
    rt=tot=lst=1;
    for(int i=1;i<=n;i++){
        int x=read();
        ins(x);
    }
    return 0;
}

4516: [Sdoi2016]生成魔咒

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 459  Solved: 282
[Submit][Status][Discuss]

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 共有多少种生成魔咒。

Input

第一行一个整数 n。

第二行 n 个数，第 i 个数表示第 i 次操作加入的魔咒字符。

1≤n≤100000。,用来表示魔咒字符的数字 x 满足 1≤x≤10^9

Output

输出 n 行，每行一个数。第 i 行的数表示第 i 次操作后 S 的生成魔咒数量

Sample Input

7
1 2 3 3 3 1 2

Sample Output

1
3
6
9
12
17
22