洛谷傳送門

題目描述

魔咒串由許多魔咒字元組成，魔咒字元可以用數字表示。例如可以將魔咒字元 $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]$

輸入輸出格式

輸入格式：

第一行一個整數 $n$。

第二行 $n$ 個數，第 $i$ 個數表示第 $i$ 次操作加入的魔咒字元。

輸出格式：

輸出 $n$ 行，每行一個數。第 $i$ 行的數表示第 $i$ 次操作後 $S$ 的生成魔咒數量

輸入輸出樣例

輸入樣例#1：

7
1 2 3 3 3 1 2

輸出樣例#1：

1
3
6
9
12
17
22

說明

對於10%的資料，$1 \le n \le 10$

對於30%的資料，$1 \le n \le 100$

對於60%的資料，$1 \le n \le 1000$

對於100%的資料，$1 \le n \le 100000$

用來表示魔咒字元的數字 $x$ 滿足$1 \le n \le 10^9$

解題分析

動態維護本質不同的子串數， 實際上就是$SAM$上每個點的$len$減去其$parent$節點的$len$的和。 我們在修改$parnet$的時候順帶維護即可。

程式碼如下：

#include <cstdio>
#include
 
 <cstring>
#include <cstdlib>
#include <cctype>
#include <algorithm>
#include <cmath>
#include <map>
#define R register
#define IN inline
#define W while
#define MX 505000
#define ll long long
int par[MX], len[MX];
std::map <int, int> to[MX];
int last, cur, l, cnt;
ll ans;
namespace SAM
{
	IN int get(R int now) {return (~par[now]) ? len[now] - len[par[now]] : 0;}
	IN void insert(R int id)
	{
		R int now = last, tar;
		cur = ++cnt; len[cur] = len[last] + 1; last = cur;
		for (; (~now) && !to[now][id]; now = par[now]) to[now][id] = cur;
		if(now < 0) return par[cur] = 0, ans += get(cur), void();
		tar = to[now][id];
		if(len[tar] == len[now] + 1) return par[cur] = tar, ans += get(cur), void();
		int nw = ++cnt; len[nw] = len[now] + 1; 
		par[nw] = par[tar]; ans += get(nw) - get(tar);
		par[tar] = par[cur] = nw; ans += get(cur) + get(tar);
		to[nw] = to[tar];
		for (; (~now) && to[now][id] == tar; now = par[now]) to[now][id] = nw;
	}
}
int main(void)
{
	par[0] = -1;
	int T, buf; scanf("%d", &T);
	W (T--)
	{
		scanf("%d", &buf);
		SAM::insert(buf); printf("%lld\n", ans);
	}
}