bzoj4516 [Sdoi2016]生成魔咒

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

 

正解：后缀自动机。

因为题目是动态从前往后加字符，所以我们使用后缀自动机正好可以解决这个问题。

然后每次增加字符以后，不同子串的增量就是当前状态到它父亲的状态的$len$的差。

 

#include <bits/stdc++.h>
#define il inline
#define RG register
#define ll long long
#define N (200010)

using namespace std;

map<int,int> ch[N];

int fa[N],l[N],n,la,tot;
ll ans;

il int gi(){
  RG int x=0,q=1; RG char ch=getchar();
  while ((ch<'0' || ch>'9') && ch!='-') ch=getchar();
  if (ch=='-') q=-1,ch=getchar();
  while (ch>='0' && ch<='9') x=x*10+ch-48,ch=getchar();
  return q*x;
}

il void add(RG int c){
  RG int p=la,np=++tot; la=tot,l[np]=l[p]+1;
  for (;p && !ch[p][c];p=fa[p]) ch[p][c]=np;
  if (!p){ fa[np]=1,ans+=l[np]-l[fa[np]]; return; }
  RG int q=ch[p][c];
  if (l[q]==l[p]+1) fa[np]=q; else{
    RG int nq=++tot; l[nq]=l[p]+1;
    fa[nq]=fa[q],fa[q]=fa[np]=nq,ch[nq]=ch[q];
    for (;ch[p][c]==q;p=fa[p]) ch[p][c]=nq;
  }
  ans+=l[np]-l[fa[np]]; return;
}

int main(){
#ifndef ONLINE_JUDGE
  freopen("string.in","r",stdin);
  freopen("string.out","w",stdout);
#endif
  n=gi(),tot=la=1;
  for (RG int i=1,c;i<=n;++i)
    c=gi(),add(c),printf("%lld\n",ans);
  return 0;
}