sdoi2016生成魔咒

4516: [Sdoi2016]生成魔咒

Time Limit: 10 Sec Memory Limit: 128 MB
Submit: 1532 Solved: 885
[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

维护答案的错误写法

1 for(int j=lst+1;j<= cnt;++j)ans+=(l[j]-l[f[j]]);
2 lst=cnt;

毕竟每次插入字符时之前的父子关系、转移关系会改变啦

 1 /**************************************************************
 2     Problem: 4516
 3     User: xln
 4     Language: C++
 5     Result: Accepted
 6     Time:820 ms
 7     Memory:15720 kb
 8 ****************************************************************/
 9  
10 #include<cstdio> 
11 #include<iostream>
12 #include<algorithm>
13 #include<map>
14 using namespace std;
15 const int N=200005;
16 map<int,int>c[N];
17 int pre=1,cnt=1,f[N],l[N],s[N],n;
18 typedef long long ll;
19 ll ans;
20 inline int gi(){int d=0;char c=getchar();while(c<'0'||c>'9')c=getchar();
21 while(c<='9'&&c>='0'){d=(d<<3)+(d<<1)+c-'0';c=getchar();}return d;}
22 inline void ins(int x){
23     int p=pre,np=++cnt;pre=np;l[np]=l[p]+1;
24     for(;p&&!c[p].count(x);p=f[p])c[p][x]=np;
25     if(!p)f[np]=1;
26     else{
27         int q=c[p][x];
28         if(l[q]==l[p]+1)f[np]=q;
29         else{
30             int nq=++cnt;
31             c[nq]=c[q];
32             f[nq]=f[q];
33             l[nq]=l[p]+1;
34             ans-=(l[q]-l[f[q]]);
35             f[q]=f[np]=nq;
36             ans+=(l[q]-l[f[q]]);
37             ans+=(l[nq]-l[f[nq]]);
38             for(;c[p][x]==q;p=f[p])c[p][x]=nq;
39         }
40     }
41     ans+=(l[np]-l[f[np]]);
42 }
43 int main(){
44     n=gi();
45     for(int i=1;i<=n;++i){
46         s[i]=gi();
47         ins(s[i]);
48         printf("%lld
",ans);
49     }
50     return 0;
51 }