[HNOI2019]JOJO

题目链接
题意:有一个字符串,初始为空,n次操作,每次可以添加一段字符到末尾或回到第x次操作之后的状态。每次操作后,输出所有前缀的next之和。
首先,那个撤销操作可以离线,建版本树解决。
既然求的是最长公共前后缀,自然想到用KMP。
模仿KMP的过程:记录每次添加后的串的next,中间的next无需记录。
然后,考虑添加一段字符:
和KMP一样,沿next链走,不过这里的匹配需要匹配一串字符,所以求出从\(i\)位置向后,能匹配\(c\)多少。
当这个长度增加时,对答案的贡献是一段等差数列。
若一个位置向后,正好能匹配\(x\)个,则将这个位置赋给添加后的next。
注意:如果匹配的个数大于\(x\)个,则不要做此操作。
理由:因为匹配的个数\(x\)个,而题目保证相邻两次的\(c\)不同,所以匹配到中间是无法往下匹配的。
所以,不会有记录的next指向中间,中间的next就无需记录。
这样,就简单多了。
但是,KMP的复杂度是均摊的,不能可持久化(会被卡超时)。
考虑一个优化:
若最长公共前后缀的长度大于串的一半,则说明出现了循环节,那样可以直接跳到第一个循环节的对应位置。
这样,无论如何,走一次next,长度至少会缩减一半。
这样,时间复杂度就是严格的\(O(nlogn)\)的了。
详见代码:

#include <stdio.h> 
#define md 998244353 
struct SJd {
	int l,r,c;
};
int getsum(int x, int y) {
	return 1ll * (x - y + 1 + x) * y / 2 % md;
}
struct String {
	SJd sz[100010];
	int ne[100010],cd;
	String() {
		cd = 0;
		sz[0].l = sz[0].r = 0;
		ne[0] = -1;
	}
	int getne(int x) {
		if (ne[x] == -1 || sz[ne[x]].r * 2 <= sz[x].r + 1) return ne[x];
		else return ne[x] % (x - ne[x]);
	}
	int insert(int x, int c) {
		cd += 1;
		sz[cd].l = sz[cd - 1].r + 1;
		sz[cd].r = sz[cd].l + x - 1;
		sz[cd].c = c;
		ne[cd] = 0;
		if (cd == 1) return getsum(x - 1, x);
		int p = getne(cd - 1),ma = 0,rt = 0,
		tp = ne[cd - 1];
		while (p != -1) {
			int t = (sz[tp + 1].c == c ? sz[tp + 1].r - sz[tp + 1].l + 1 : 0);
			bool b = false;
			if (t > x) {
				t = x;
				b = true;
			}
			if (t > ma) {
				rt = (rt + getsum(sz[tp].r + t, t - ma)) % md;
				ma = t;
			}
			if (t == x && !b) {
				ne[cd] = tp + 1;
				break;
			}
			p = getne(p);
			tp = ne[tp];
		}
		if (ma < x && sz[1].c == c) {
			rt = (rt + 1ll * (x - ma) * (sz[1].r - sz[1].l + 1)) % md;
			ne[cd] = 1;
		}
		return rt;
	}
};
String str;
int tm[100010],ans[100010];
int fr[100010],ne[100010],v[100010],bs = 0;
int lx[100010],x[100010],c[100010];
void addb(int a, int b) {
	v[bs] = b;
	ne[bs] = fr[a];
	fr[a] = bs++;
}
void dfs(int u, int he) {
	int oldcd = str.cd;
	if (u != 0) he = (he + str.insert(x[u], c[u])) % md;
	ans[u] = he;
	for (int i = fr[u]; i != -1; i = ne[i]) dfs(v[i], he);
	str.cd = oldcd;
}
int main() {
	int n;
	scanf("%d", &n);
	for (int i = 0; i <= n; i++) fr[i] = -1;
	for (int i = 1; i <= n; i++) {
		int lx;
		scanf("%d", &lx);
		if (lx == 1) {
			char ch[2];
			scanf("%d%s", &x[i], ch);
			c[i] = ch[0] - 'a';
			addb(tm[i - 1], i);
			tm[i] = i;
		} else {
			int a;
			scanf("%d", &a);
			tm[i] = tm[a];
		}
	}
	dfs(0, 0);
	for (int i = 1; i <= n; i++) {
		if (lx[i] == 1) printf("%d\n", ans[i]);
		else printf("%d\n", ans[tm[i]]);
	}
	return 0;
}
posted @ 2019-08-13 20:42  lnzwz  阅读(178)  评论(0编辑  收藏  举报