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

4516: [Sdoi2016]生成魔咒

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 1328  Solved: 745

[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

考虑每次加入一个字符，可能产生的新子串一定是一个后缀

所以我们只要每次找到新建节点的parent

其mx就是不会产生的新后缀数 用串长减掉即可

突然发现个nq赋状态的时候写了个迭代器。。太傻了。。。

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<set>
#include<map>
using namespace std;

typedef long long ll;

inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
void print(ll x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=200100;

struct SAM
{
map<int,int> trans
;
int par
,mx
;
int sz,suff,root;

SAM(){sz=suff=root=1;}

int insert(int x)
{
int p=suff,np=++sz;
mx[np]=mx[p]+1;
while(p && !trans[p][x])
trans[p][x]=np,p=par[p];
if(!p) par[np]=root;
else
{
int q=trans[p][x];
if(mx[q]==mx[p]+1) par[np]=q;
else
{
int nq=++sz;
mx[nq]=mx[p]+1;
map<int,int>::iterator it;
for(it=trans[q].begin();it!=trans[q].end();++it)
trans[nq][it->first]=it->second;
par[nq]=par[q];
par[q]=par[np]=nq;
while(p && trans[p][x]==q)
trans[p][x]=nq,p=par[p];
}
}
suff=np;
return mx[par[np]];
}
}sam;

int main()
{
int n=read();
ll ans(0);
register int i,x;
for(i=1;i<=n;++i)
{
x=read();
ans+=i-sam.insert(x);
print(ans);putchar('\n');
}
return 0;
}