BZOJ4943 NOI2017蚯蚓排队（哈希+链表）

　　能看懂题就能想到正解。维护所有长度不超过k的数字串的哈希值即可，用链表维护一下蚯蚓间连接情况。由于这样的数字串至多只有nk个，计算哈希值的总复杂度为O(nk)，而分裂的复杂度为O(ck^2)，询问复杂度为O(Σ|s|)。于是总复杂度为O(nk+ck^2+Σ|s|)。

　　手写哈希注意插入元素时考虑清楚，如果没有哈希冲突不需要更新哈希使用的链表，所以特判一下。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
int read()
{
    int x=;char c=getchar();
    while (c<''||c>'') c=getchar();
    while (c>=''&&c<='') x=(x<<)+(x<<)+(c^),c=getchar();
    return x;
}
#define N 200010
#define S 10000010
#define K 50
#define P 19260817
#define MOD 998244353
#define G 7
#define ul unsigned long long
int n,m,POW[K+][];
struct data{int x,pre,nxt;
}a[N];
int ch[S];
int map[P+],cnt[P+],nxt[P+];
ul value[P+],POWv[K+][];
void ins(int x,ul v)
{
    if (!map[x]) map[x]=,value[x]=v,cnt[x]=;
    else
    {
        int t;
        while (map[x]&&value[x]!=v) t=x,x=nxt[x];
        if (map[x]) cnt[x]++;
        else map[x]=,value[x]=v,cnt[x]=,nxt[t]=x;
    }
}
void del(int x,ul v)
{
    while (value[x]!=v) x=nxt[x];
    cnt[x]--;
}
int query(int x,ul v)
{
    while (map[x]&&value[x]!=v) x=nxt[x];
    return cnt[x];
}
int main()
{
    n=read(),m=read();
    for (int i=;i<P-;i++) nxt[i]=i+;nxt[P-]=;
    for (int j=;j<=;j++)
    {
        POW[][j]=j;for (int i=;i<=K;i++) POW[i][j]=POW[i-][j]*G%P;
        POWv[][j]=j;for (int i=;i<=K;i++) POWv[i][j]=POWv[i-][j]*G;
    }
    for (int i=;i<=n;i++) a[i].x=read(),ins(a[i].x,a[i].x);
    while (m--)
    {
        int op=read();
        switch(op)
        {
            case :
            {
                int x=read(),y=read();
                a[x].nxt=y,a[y].pre=x;
                for (int l=,h=x;h&&l<K;h=a[h].pre,l++)
                {
                    int hash=;ul hashv=;
                    for (int i=,t=h;i<=l;t=a[t].nxt,i++)
                    hash=(hash*G+a[t].x)%P,hashv=hashv*G+a[t].x;
                    for (int i=l+,t=y;t&&i<=K;t=a[t].nxt,i++)
                    {
                        hash=(hash*G+a[t].x)%P,hashv=hashv*G+a[t].x;
                        ins(hash,hashv);
                    }
                }
                break;
            }
            case :
            {
                int x=read(),y=a[x].nxt;
                for (int l=,h=x;h&&l<K;h=a[h].pre,l++)
                {
                    int hash=;ul hashv=;
                    for (int i=,t=h;i<=l;t=a[t].nxt,i++)
                    hash=(hash*G+a[t].x)%P,hashv=hashv*G+a[t].x;
                    for (int i=l+,t=y;t&&i<=K;t=a[t].nxt,i++)
                    {
                        hash=(hash*G+a[t].x)%P,hashv=hashv*G+a[t].x;
                        del(hash,hashv);
                    }
                }
                a[x].nxt=;a[y].pre=;
                break;
            }
            case :
            {
                char c=getchar();int l=;
                while (c<''||c>'') c=getchar();
                while (c>=''&&c<='') ch[++l]=c^,c=getchar();
                int k=read();
                int hash=;ul hashv=;
                for (int i=;i<k;i++)
                hash=(hash*G+ch[i])%P,hashv=hashv*G+ch[i];
                int ans=;
                for (int i=k;i<=l;i++)
                {
                    hash=(hash*G+ch[i]-POW[k][ch[i-k]]+P)%P,
                    hashv=hashv*G+ch[i]-POWv[k][ch[i-k]];
                    ans=1ll*ans*query(hash,hashv)%MOD;
                }
                printf("%d\n",ans);
            }
        }
    }
    return ;
}