LOJ#2303. 「NOI2017」蚯蚓排队

$n leq 200000$的$1 leq a_i leq 6$的蚯蚓，有三种操作：让一只队头蚯蚓接在一只队尾蚯蚓后面；让一队蚯蚓从某个蚯蚓后面断成两队；问：给个字符串，问他的。。算了你们直接看题吧

这什么沙雕题QAQ

所有询问的串只有$nk$种，把他们全丢进hash里面就好了。。注意双hash，一个用来当链表一个用来在链表里判重。

复杂度的话，只考虑合并是$nk$的（相当于把所有串算一次），而拆分是$ck^2$的，拆对合并的复杂度影响是跟拆本身复杂度一样的。

//#include<iostream>
#include<cstring>
#include<cstdio>
//#include<math.h>
//#include<set>
//#include<queue>
//#include<bitset>
//#include<vector>
#include<algorithm>
#include<stdlib.h>
using namespace std;

#define LL long long
int qread()
{
    char c; int s=0,f=1; while ((c=getchar())<'0' || c>'9') (c=='-') && (f=-1);
    do s=s*10+c-'0'; while ((c=getchar())>='0' && c<='9'); return s*f;
}

//Pay attention to '-' , LL and double of qread!!!!

int n,m;
#define maxn 200011
#define maxh 1000007
int a[maxn],bb[55]; LL cc[55];

struct Hash
{
    struct Edge{LL to; int v,next;}edge[maxn*50]; int first[maxh],le;
    Hash() {le=2;}
    void in(LL x,int h,int v)
    {
        for (int i=first[h];i;i=edge[i].next)
        {
            Edge &e=edge[i];
            if (e.to==x) {e.v+=v; return;}
        }
        Edge &e=edge[le]; e.to=x; e.v=1; e.next=first[h]; first[h]=le++;
    }
    int find(LL x,int h)
    {
        for (int i=first[h];i;i=edge[i].next)
        {
            Edge &e=edge[i];
            if (e.to==x) return e.v;
        }
        return 0;
    }
}h;

int Nxt[maxn],Pre[maxn];
char s[maxn*50]; int len;
int main()
{
    bb[0]=1; for (int i=1;i<=50;i++) bb[i]=bb[i-1]*7%maxh;
    cc[0]=1; for (int i=1;i<=50;i++) cc[i]=cc[i-1]*7;
    n=qread(); m=qread();
    for (int i=1;i<=n;i++) {Pre[i]=Nxt[i]=0; a[i]=qread(); h.in(a[i],a[i],1);}
    int op; char c; int x,y;
    while (m--)
    {
        op=qread();
        if (op==1)
        {
            x=qread(); y=qread();
            Nxt[x]=y; Pre[y]=x;
            LL B=0,C=0;
            for (int i=x,cnt=49,w=0;cnt && i;i=Pre[i],w++,cnt--)
            {
                B=(B+1ll*a[i]*bb[w])%maxh;
                C=C+1ll*a[i]*cc[w];
                LL nb=B,nc=C;
                for (int j=y,k=1;j && k<=cnt;j=Nxt[j],k++)
                {
                    nb=(nb*7+a[j])%maxh;
                    nc=nc*7+a[j];
                    h.in(nc,nb,1);
                }
            }
        }
        else if (op==2)
        {
            x=qread(); y=Nxt[x];
            Nxt[x]=0; Pre[y]=0;
            LL B=0,C=0;
            for (int i=x,cnt=49,w=0;cnt && i;i=Pre[i],w++,cnt--)
            {
                B=(B+1ll*a[i]*bb[w])%maxh;
                C=C+1ll*a[i]*cc[w];
                LL nb=B,nc=C;
                for (int j=y,k=1;j && k<=cnt;j=Nxt[j],k++)
                {
                    nb=(nb*7+a[j])%maxh;
                    nc=nc*7+a[j];
                    h.in(nc,nb,-1);
                }
            }
        }
        else if (op==3)
        {
            len=0;
            while ((c=getchar())<'0' || c>'9');
            do s[++len]=c; while ((c=getchar())>='0' && c<='9');
            x=qread();
            LL B=0,C=0;
            for (int i=1;i<x;i++) B=(B*7+s[i]-'0')%maxh,C=C*7+s[i]-'0';
            int ans=1;
            for (int i=x;i<=len;i++)
            {
                B=(B*7+s[i]-'0')%maxh; C=C*7+s[i]-'0';
                ans=1ll*ans*h.find(C,B)%998244353;
                B=(B-(s[i-x+1]-'0')*bb[x-1])%maxh+maxh; B%=maxh;
                C=C-(s[i-x+1]-'0')*cc[x-1];
            }
            printf("%d
",ans);
        }
    }
    return 0;
}