[BZOJ 4571][scoi2016]美味 主席樹

從高到低按位貪心,對於原來的數列建立權值主席樹,每一次貪心的算出區間範圍然後主席樹查詢有沒有這個區間有沒有這個值,然後就沒有了

#include<cstdio>
#include<cstring>
#include<iostream>
#define maxn 200021
using namespace std;
int read(){
	char c=getchar();int x=0;
	for(;c>'9'||c<'0';c=getchar());
	for(;c<='9'&&c>='0';c=getchar())x=x*10+c-'0';
	return x;
}
int sum[maxn*40],ls[maxn*40],rs[maxn*40],n,m,rt[maxn],mx;
int tot,a[maxn];
void insert(int x,int& y,int l,int r,int id){
	y=++tot;ls[y]=ls[x],rs[y]=rs[x],sum[y]=sum[x]+1;
	if(l==r)return;
	int mid=l+r>>1;
	if(id>mid)insert(rs[x],rs[y],mid+1,r,id);
	else insert(ls[x],ls[y],l,mid,id);
}
int query(int x,int y,int l,int r,int s,int t){
	if(l==s&&r==t)return sum[y]-sum[x];
	int mid=l+r>>1;
	if(s>mid)return query(rs[x],rs[y],mid+1,r,s,t);
	else if(t<=mid)return query(ls[x],ls[y],l,mid,s,t);
	else return query(ls[x],ls[y],l,mid,s,mid)+
				query(rs[x],rs[y],mid+1,r,mid+1,t);
}
int main(){
	n=read(),m=read();
	for(int i=1;i<=n;i++){
		a[i]=read();mx=max(a[i],mx);
	}
	for(int i=1;i<=n;i++)insert(rt[i-1],rt[i],0,mx,a[i]);
	int l,r,x,b;
	while(m--){
		b=read(),x=read(),l=read(),r=read();
		int ans=0,s,t;
		for(int c,i=19;i>=0;i--){
			if(b>>i&1){
				s=ans,t=ans|((1<<i)-1);
				ans^=1<<i;
			}
			else s=ans|(1<<i),t=ans|((1<<i+1)-1);
			s=max(0,s-x),t=min(t-x,mx);
			if(s<=t&&query(rt[r],rt[l-1],0,mx,s,t))ans^=1<<i;
		}printf("%d\n",ans^b);
	}
	return 0;
}