bzoj4943: [Noi2017]蚯蚓

解题思路：

感觉这题巨水，k很小，每次操作最多会产生或减少k^2个字符串，直接用哈希表暴力统计贡献就行了。难怪uoj差评……

#include<bits/stdc++.h>
#define ll unsigned long long
using namespace std;
int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')c=getchar(),f=-1;
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}
const int N=200005,M=10000010,H=1e7+9,base=7,mod=998244353;
int n,m,c[N],l[N],r[N];
int tot,first[M],nxt[M],cnt[M];ll val[M];
ll h[M],bin[51];
char s[M];
void add(ll x)
{
    int y=x%H;
    for(int v=first[y];v;v=nxt[v])
        if(val[v]==x){cnt[v]++;return;}
    nxt[++tot]=first[y],first[y]=tot,val[tot]=x,cnt[tot]=1;
}
void del(ll x)
{
    int y=x%H;
    for(int v=first[y];v;v=nxt[v])
        if(val[v]==x){cnt[v]--;return;}
}
int query(ll x)
{
    int y=x%H;
    for(int v=first[y];v;v=nxt[v])
        if(val[v]==x)return cnt[v];
    return 0;
}
ll cut(int l,int r)
{
    return h[r]-h[l-1]*bin[r-l+1];
}
int main()
{
    //freopen("lx.in","r",stdin);
    bin[0]=1;for(int i=1;i<=50;i++)bin[i]=bin[i-1]*base;
    n=getint(),m=getint();
    for(int i=1;i<=n;i++)add(c[i]=getint());
    while(m--)
    {
        int op=getint();
        if(op==1)
        {
            int x=getint(),y=getint();
            r[x]=y,l[y]=x;
            for(int i=1;i<=50&&x;i++,x=l[x])
            {
                ll cur=0;int len=0,j=x;
                for(;j!=y;j=r[j])++len,cur=cur*base+c[j];
                for(++len;len<=50&&j;len++,j=r[j])cur=cur*base+c[j],add(cur);
            }
        }
        else if(op==2)
        {
            int x=getint(),y=r[x];
            for(int i=1;i<=50&&x;i++,x=l[x])
            {
                ll cur=0;int len=0,j=x;
                for(;j!=y;j=r[j])++len,cur=cur*base+c[j];
                for(++len;len<=50&&j;len++,j=r[j])cur=cur*base+c[j],del(cur);
            }
            x=l[y],r[x]=l[y]=0;
        }
        else 
        {
            scanf("%s",s+1);
            int k=getint(),len=strlen(s+1);
            for(int i=1;i<=len;i++)h[i]=h[i-1]*base+s[i]-'0';
            ll ans=1;
            for(int i=1;i<=len-k+1;i++)ans=ans*query(cut(i,i+k-1))%mod;
            printf("%lld\n",ans);
        }
    }
    return 0;
}