Description
蚯蚓幼儿园有 \(n\) 只蚯蚓。幼儿园园长神刀手为了管理方便,时常让这些蚯蚓们列队表演。
所有蚯蚓用从 \(1\) 到 \(n\) 的连续正整数编号。每只蚯蚓的长度可以用一个正整数表示,根据入园要求,所有蚯蚓的长度都不超过 \(6\) 。神刀手希望这些蚯蚓排成若干个队伍,初始时,每只蚯蚓各自排成一个仅有一只蚯蚓的队伍,该蚯蚓既在队首,也在队尾。
神刀手将会依次进行 \(m\) 次操作,每个操作都是以下三种操作中的一种:
1.给出 \(i\) 和 \(j\) ,令 \(i\) 号蚯蚓与 \(j\) 号蚯蚓所在的两个队伍合并为一个队伍,具体来说,令 \(j\) 号蚯蚓紧挨在 \(i\) 号蚯蚓之后,其余蚯蚓保持队伍的前后关系不变。
2.给出 \(i\) ,令 \(i\) 号蚯蚓与紧挨其后的一只蚯蚓分离为两个队伍,具体来说,在分离之后, \(i\) 号蚯蚓在其中一个队伍的队尾,原本紧挨其后的那一只蚯蚓在另一个队伍的队首,其余蚯蚓保持队伍的前后关系不变。
3.给出一个正整数 \(k\) 和一个长度至少为 \(k\) 的数字串 \(s\) ,对于 \(s\) 的每个长度为 \(k\) 的连续子串 \(t\) (这样的子串共有 \(|s| - k + 1\) 个,其中 \(|s|\) 为 \(s\) 的长度),定义函数 \(f(t)\) ,询问所有这些 \(f(t)\) 的乘积对 \(998244353\) 取模后的结果。其中 \(f(t)\) 的定义如下: 对于当前的蚯蚓队伍,定义某个蚯蚓的向后 \(k\) 数字串为:从该蚯蚓出发,沿队伍的向后方向,寻找最近的 \(k\) 只蚯蚓(包括其自身),将这些蚯蚓的长度视作字符连接而成的数字串;如果这样找到的蚯蚓不足 \(k\) 只,则其没有向后 \(k\) 数字串。例如蚯蚓的队伍为 \(10\) 号蚯蚓在队首,其后是 \(22\) 号蚯蚓,其后是 \(3\) 号蚯蚓(为队尾),这些蚯蚓的长度分别为 \(4\) 、 \(5\) 、 \(6\) ,则 \(10\) 号蚯蚓的向后 \(3\) 数字串 为456, \(22\) 号蚯蚓没有向后 \(3\) 数字串,但其向后 \(2\) 数字串为56,其向后 \(1\) 数字串为5。
而 \(f(t)\) 表示所有蚯蚓中,向后 \(k\) 数字串恰好为 \(t\) 的蚯蚓只数。
保证 \(n \leqslant 2 \times 10^5,\ m \leqslant 5 \times 10^5,\ k \leqslant 50\) 。设 \(\sum |S|\) 为某个输入文件中所有询问的 s 的长度总和,则 \(\sum |S| \leqslant 10^7\) 。 设 \(c\) 为某个输入文件中形如 2 i
的操作的次数,则 \(c \leqslant 10^3\) 。
Solution
由于 \(k\) 比较小,直接 \(hash\) 就行了。复杂度是正确的,因为整个字符串需被 \(hash\) 的子串是 \(O(nk)\) 的。由于 \(c\leq 10^3\) ,所以修改的字符串复杂度是 \(O(ck^2)\) 的。
总复杂度是 \(O\left(nk+ck^2+\sum|S|\right)\) 。
Code
#include <bits/stdc++.h>
#define ull unsigned long long
using namespace std;
const int N = 2e5+5, LEN = 1e7+5, yzh = 998244353, p = 12456791, base = 31;
void gi(int &x) {
char ch = getchar(); x = 0;
for (; ch < '0' || ch > '9'; ch = getchar());
for (; ch >= '0' && ch <= '9'; ch = getchar()) x = (x<<1)+(x<<3)+ch-48;
}
char ch[LEN];
int n, m, l[N], lst[N], nxt[N], opt, x, y, k;
ull bin[N], tmp;
struct hash_table {
int cnt[p]; ull ky[p];
void insert(ull x, int val) {
int loc = x%p;
while (true) {
if (ky[loc] == 0 || ky[loc] == x) {cnt[loc] += val, ky[loc] = x; break; }
++loc; if (loc == p) loc = 0;
}
}
int count(ull x) {
int loc = x%p;
while (true) {
if (ky[loc] == 0 || ky[loc] == x) return cnt[loc];
++loc; if (loc == p) loc = 0;
}
}
}mp;
void merge() {
int t = 48, loc = x;
while (t-- && lst[loc]) loc = lst[loc];
lst[y] = x, nxt[x] = y;
for (; lst[loc] != x; loc = nxt[loc]) {
bool flag = 0; t = loc; tmp = 0;
for (int j = 1; j <= 50 && t; j++, t = nxt[t]) {
tmp = tmp*base+l[t];
if (flag) mp.insert(tmp, 1);
if (t == x) flag = 1;
}
}
}
void split() {
int t = 48, loc = x; y = nxt[x];
while (t-- && lst[loc]) loc = lst[loc];
for (; lst[loc] != x; loc = nxt[loc]) {
bool flag = 0; t = loc; tmp = 0;
for (int j = 1; j <= 50 && t; j++, t = nxt[t]) {
tmp = tmp*base+l[t];
if (flag) mp.insert(tmp, -1);
if (t == x) flag = 1;
}
}
lst[y] = nxt[x] = 0;
}
int query() {
int n = strlen(ch+1), ans = 1; tmp = 0;
for (int i = 1; i <= n; i++) {
tmp = tmp*base+ch[i]-48;
if (i > k) tmp -= 1ll*(ch[i-k]-48)*bin[k];
if (i >= k) ans = 1ll*ans*mp.count(tmp)%yzh;
}
return ans;
}
void work() {
gi(n), gi(m); bin[0] = 1;
for (int i = 1; i <= n; i++) gi(l[i]), mp.insert(l[i], 1);
for (int i = 1; i <= 50; i++) bin[i] = bin[i-1]*base;
while (m--) {
gi(opt);
if (opt == 1) gi(x), gi(y), merge();
else if (opt == 2) gi(x), split();
else scanf("%s", ch+1), gi(k), printf("%d\n", query());
}
}
int main() {
freopen("queue.in", "r", stdin);
freopen("queue.out", "w", stdout);
work(); return 0;
}