[主席树维护HASH] Codechef. Cloning

传送门

直接用主席树维护HASH比较就可以了吧…

#include <cstdio>
#include <algorithm>
#include <iostream>

using namespace std;

typedef unsigned long long ll;

const int N=100010;
const ll base=100003;

int t,n,q,cnt;
int a[N],rt[N],ls[N*50],rs[N*50],tot[N*50];
ll v[N*50],p[N];

void Add(int &g,int l,int r,int x){
  int k=g; g=++cnt; ls[g]=ls[k]; rs[g]=rs[k]; v[g]=v[k]; tot[g]=tot[k];
  if(l==r){
    v[g]++; tot[g]++; return ;
  }
  int mid=l+r>>1;
  if(x<=mid) Add(ls[g],l,mid,x); else Add(rs[g],mid+1,r,x);
  v[g]=v[ls[g]]*p[r-mid]+v[rs[g]]; tot[g]=tot[ls[g]]+tot[rs[g]];
}

inline int dif(int x,int y,int a,int b){
  if(v[y]-v[x]==v[b]-v[a]) return 0;
  int l=1,r=n;
  while(l<r){
    int mid=l+r>>1;
    if(v[ls[y]]-v[ls[x]]==v[ls[b]]-v[ls[a]])
      x=rs[x],y=rs[y],a=rs[a],b=rs[b],l=mid+1;
    else
      x=ls[x],y=ls[y],a=ls[a],b=ls[b],r=mid;
  }
  return l;
}

int Find(int x,int y,int L,int R,int l,int r){
  if(v[x]-v[y]==0) return -1;
  if(L==R) return tot[x]-tot[y]==0?-1:L;
  int mid=L+R>>1;
  if(l==L && r==R){
    if(v[ls[x]]-v[ls[y]]) return Find(ls[x],ls[y],L,mid,l,mid);
    else return Find(rs[x],rs[y],mid+1,R,mid+1,r);
  }
  if(r<=mid) return Find(ls[x],ls[y],L,mid,l,r);
  else if(l>mid) return Find(rs[x],rs[y],mid+1,R,l,r);
  int ret=Find(ls[x],ls[y],L,mid,l,mid);
  if(~ret) return ret;
  return Find(rs[x],rs[y],mid+1,R,mid+1,r);
}

int Check(int x,int y,int L,int R,int p){
  if(L==R) return tot[x]-tot[y];
  int mid=L+R>>1;
  if(p<=mid) return Check(ls[x],ls[y],L,mid,p);
  else return Check(rs[x],rs[y],mid+1,R,p);
}

int main(){
  scanf("%d",&t);
  p[0]=1; for(int i=1;i<=100000;i++) p[i]=p[i-1]*base;
  while(t--){
    scanf("%d%d",&n,&q); cnt=0;
    for(int i=1;i<=n;i++) rt[i]=0;
    for(int i=1;i<=n;i++){
      scanf("%d",&a[i]); rt[i]=rt[i-1];
      Add(rt[i],1,n,a[i]);
    }
    for(int i=1;i<=q;i++){
      int l1,r1,l2,r2;
      scanf("%d%d%d%d",&l1,&r1,&l2,&r2);
      int pos=dif(rt[l1-1],rt[r1],rt[l2-1],rt[r2]);
      if(!pos){
    puts("YES");
    continue;
      }
      int A=Check(rt[r1],rt[l1-1],1,n,pos),B=Check(rt[r2],rt[l2-1],1,n,pos);
      if(A>B){
    A=Find(rt[r1],rt[l1-1],1,n,pos,n); B=Find(rt[r2],rt[l2-1],1,n,pos+1,n);
      }
      else{
    A=Find(rt[r1],rt[l1-1],1,n,pos+1,n); B=Find(rt[r2],rt[l2-1],1,n,pos,n);
      }
      int curA=rt[r1],curB=rt[r2],lst=cnt;
      if(~A) Add(curB,1,n,A);
      if(~B) Add(curA,1,n,B);
      if(!dif(rt[l1-1],curA,rt[l2-1],curB)) puts("YES");
      else puts("NO");
      cnt=lst;
    }
  }
  return 0;
}