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

2022-11-11 22:08:39

#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

码农公寓