bzoj 4516: [Sdoi2016]生成魔咒 — 后缀自动机
Contents
4516: [Sdoi2016]生成魔咒
Time Limit: 10 Sec Memory Limit: 128 MB
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
1 2 3 3 3 1 2
Sample Output
1
3
6
9
12
17
22
3
6
9
12
17
22
HINT
Source
板子题。。。
#include<map>
#include<cmath>
#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define inf 1000000007
#define ll long long
#define N 200010
inline int rd()
{
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 n;
ll ans;
int fa[N],l[N],lst,cnt;
map<int,int>c[N];
void ins(int x)
{
int p,q,np,nq;
p=lst;np=lst=++cnt;
l[np]=l[p]+1;
while(!c[p][x]&&p) c[p][x]=np,p=fa[p];
if(!p) fa[np]=1;
else
{
q=c[p][x];
if(l[q]==l[p]+1) fa[np]=q;
else
{
nq=++cnt;l[nq]=l[p]+1;
c[nq]=c[q];
fa[nq]=fa[q];
fa[q]=fa[np]=nq;
while(c[p][x]==q) c[p][x]=nq,p=fa[p];
}
}
ans+=l[np]-l[fa[np]];
}
int main()
{
n=rd();
lst=cnt=1;
for(int i=1,x;i<=n;i++)
{
x=rd();
ins(x);
printf("%lld\n",ans);
}
return 0;
}
发表评论