bzoj 3207 花神的嘲讽计划Ⅰ（哈希法+主席树）

【题目链接】

    http://www.lydsy.com/JudgeOnline/problem.php?id=3207

【题意】

    给定一个文本串，多次询问K长的模式串是否在文本[l,r]区间内出现。

【思路】

    对文本串每K个长度进行一次hash。

    主席树维护val表示一个hash值 的出现次数。查询只需要找到对应的两棵主席树然后计算val的差值即可。

【代码】

#include<set>
#include<cmath>
#include<queue>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define trav(u,i) for(int i=front[u];i;i=e[i].nxt)
#define FOR(a,b,c) for(int a=(b);a<=(c);a++)
using namespace std;

typedef unsigned long long ll;
const int N = 2e5+10;

ll read() {
    char c=getchar();
    ll f=1,x=0;
    while(!isdigit(c)) {
        if(c=='-') f=-1; c=getchar();
    }
    while(isdigit(c))
        x=x*10+c-'0',c=getchar();
    return x*f;
}

struct Tnode {
    Tnode *ls,*rs;
    int val;
} *T[N],mempool[N*50],*G=mempool;
Tnode * Nw (Tnode*l,Tnode*r,int x) {
    G->ls=l,G->rs=r,G->val=x;
    return G++;
}

Tnode* build(Tnode* p,ll l,ll r,ll pos) {
    if(l==r) return Nw (0x0,0x0,p->val+1);
    else {
        ll mid=l+((r-l)>>1);
        if(pos<=mid) return Nw (build(p->ls,l,mid,pos),p->rs,p->val);
        else return Nw (p->ls,build(p->rs,mid+1,r,pos),p->val);
    }
}
int query(Tnode*x,Tnode*y,ll l,ll r,ll pos) {
    if(l==r) return y->val - x->val;
    else {
        ll mid=l+((r-l)>>1);
        if(pos<=mid) return query(x->ls,y->ls,l,mid,pos);
        else return query(x->rs,y->rs,mid+1,r,pos);
    }
}

int n,m,K,a;
ll H[N],xp=1,X=233;

int main()
{
//    freopen("in.in","r",stdin);
//    freopen("out.out","w",stdout);
    n=read(),m=read(),K=read();
    FOR(i,1,n) {
        a=read();
        H[i]=H[i-1]*X+a;
    }
    FOR(i,1,K) xp*=X;
    int l,r;
    
    T[K-1]=Nw (G,G,0);
    FOR(i,K,n)
        T[i]=build(T[i-1],0ull,18446744073709551615ull,H[i]-H[i-K]*xp);
    FOR(i,1,m) {
        l=read(),r=read();
        ll H=0;
        FOR(j,1,K) {
            a=read();
            H=H*X+a;
        }
        if(r-l+1<K) { puts("Yes"); continue; }
        if(query(T[l+K-2],T[r],0ull,18446744073709551615ull,H) > 0) puts("No");
        else puts("Yes");
    }
    return 0;
}

P.S.原来hash函数写的丑也是会WA的 TAT