洛谷P3832 [NOI2017]蚯蚓排队 【链表 + 字符串hash】

题目链接

洛谷P3832

题解

字符串哈希然后丢到hash表里边查询即可
因为\(k \le 50\)，1、2操作就暴力维护一下

经复杂度分析会发现直接这样暴力维护是对的

一开始自然溢出WA了，还以为是哈希冲突，改成双哈希后依旧WA
后来才发现是sb了漏了一句QAQ
不卡自然溢出

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define LL long long int
#define ULL unsigned long long int
#define REP(i,n) for (register int i = 1; i <= (n); i++)
#define res register
using namespace std;
const int maxn = 200005,maxh = 10000005,INF = 1000000000;
const int md = 998244353,P = 19260817;
inline int read(){
res int out = 0,flag = 1; res char c = getchar();
while (c < 48 || c > 57){if (c == '-') flag = -1; c = getchar();}
while (c >= 48 && c <= 57){out = (out << 3) + (out << 1) + c - 48; c = getchar();}
return out * flag;
}
inline void write(int x){
if (x / 10) write(x / 10);
putchar(x % 10 + '0');
}
struct Hash_Chart{
ULL num[maxh];
int head[P + 5],nxt[maxh],cnt,sum[maxh];
void ins(ULL x){
int id = x % P;
if (!head[id]){
head[id] = ++cnt;
num[cnt] = x;
sum[cnt] = 1;
return;
}
for (res int k = head[id]; k; k = nxt[k]){
if (num[k] == x) {sum[k]++; return;}
if (!nxt[k]){
nxt[k] = ++cnt;
num[cnt] = x;
sum[cnt] = 1;
return;
}
}
}
void del(ULL x){
int id = x % P;
for (res int k = head[id]; k; k = nxt[k])
if (num[k] == x) {sum[k]--; return;}
}
int find(ULL x){
int id = x % P;
for (res int k = head[id]; k; k = nxt[k]){
if (num[k] == x) return sum[k];
}
return 0;
}
}hash;
ULL bin[maxn],num[maxn],s1[maxn],s2[maxn];
int lk[maxn],rk[maxn],t1,t2,n,m;
char s[maxh];
inline void solve1(){
int u = read(),v = read();
t1 = 0;
for (res int k = u; k && t1 < 50; k = lk[k]){
++t1;
s1[t1] = s1[t1 - 1] + bin[t1 - 1] * num[k];
}
t2 = 0;
for (res int k = v; k && t2 < 50; k = rk[k]){
++t2;
s2[t2] = s2[t2 - 1] * 7 + num[k];
}
for (res int i = 1; i <= t1; i++)
for (res int j = 1; j <= t2 && i + j <= 50; j++)
hash.ins(s2[j] + s1[i] * bin[j]);
rk[u] = v; lk[v] = u;
}
inline void solve2(){
int u = read(),v = rk[u];
t1 = 0;
for (res int k = u; k && t1 < 50; k = lk[k]){
++t1;
s1[t1] = s1[t1 - 1] + bin[t1 - 1] * num[k];
}
t2 = 0;
for (res int k = v; k && t2 < 50; k = rk[k]){
++t2;
s2[t2] = s2[t2 - 1] * 7 + num[k];
}
for (res int i = 1; i <= t1; i++)
for (res int j = 1; j <= t2 && i + j <= 50; j++)
hash.del(s2[j] + s1[i] * bin[j]);
rk[u] = lk[v] = 0;
}
inline void solve3(){
scanf("%s",s + 1); s[0] = '0';
int len = strlen(s + 1),k = read();
LL ans = 1; ULL h = 0;
for (res int i = 1; i <= len; i++){
h = h * 7 + s[i] - '0';
if (i >= k){
h -= bin[k] * (s[i - k] - '0');
ans = ans * hash.find(h) % md;
}
}
write(ans); putchar('\n');
}
int main(){
bin[0] = 1; for (res int i = 1; i < maxn; i++) bin[i] = bin[i - 1] * 7;
n = read(); m = read(); int opt;
REP(i,n) hash.ins(num[i] = read());
while (m--){
opt = read();
if (opt == 1) solve1();
else if (opt == 2) solve2();
else solve3();
}
return 0;
}