bzoj 3207 花神的嘲讽计划Ⅰ 主席树

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define UL unsigned long long
#define inf 18446744073709551615UL
const int N=8000005;
struct node{
int ls,rs,size;
}tr
;
int cnt,n,m,k,rt[N/40];
UL bse,w[N/40],Hash[N/40];
void insert(int &x,UL l,UL r,UL pos){
tr[++cnt]=tr[x];x=cnt;tr[x].size++;
if(l==r) return;
UL mid=l/2+r/2;if((l&1)&&(r&1)) mid++;
if(pos<=mid) insert(tr[x].ls,l,mid,pos);
else insert(tr[x].rs,mid+1,r,pos);
}
int ask(int pre,int x,UL l,UL r,UL k){
if(l==r) return tr[x].size-tr[pre].size;
UL mid=l/2+r/2;if((l&1)&&(r&1)) mid++;
if(k<=mid) return ask(tr[pre].ls,tr[x].ls,l,mid,k);
else return ask(tr[pre].rs,tr[x].rs,mid+1,r,k);
}
int main(){
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=n;i++) scanf("%d",&w[i]);
for(int i=1;i<=n;i++) Hash[i]=Hash[i-1]*107+w[i];
bse=1;
for(int i=1;i<=k;i++) bse*=107;
for(int i=k;i<=n;i++) {rt[i]=rt[i-1];
insert(rt[i],1,inf,Hash[i]-Hash[i-k]*bse);}
int x,y,_val;
while(m--){
scanf("%d%d",&x,&y);bse=0;
for(int i=1;i<=k;i++){scanf("%d",&_val);bse=bse*107+_val;}
if(ask(rt[x+k-2],rt[y],1,inf,bse)) puts("No");
else puts("Yes");
}
return 0;
}