hdu6703 2019CCPC网络选拔赛 1002 array

我对主席树的理解再次升华了

不难看出对于每次2操作,答案最大是n+1(因为每次更新是+10000000, 永远不会占用n+1,而且k是保证<=n的). 如果某个数字被进行过1操作, 那么就代表这个数字可以用于2类操作的查询,把这个数字加到set里.对于找2类询问的答案,第一种是从a[r+1,n]里找>=k的最小值(主席树实现),第二种是从这个set里找>=k的最小值(对set进行二分查找),取两者间最小值即可.

更新:

举个例子 :n=6,a={6 2 4 3 5 1} 

操作依次为 (1,3) (2,4,4) (1,5) (1,1,5)以上值均是异或lastans后的值

对于(1,3), 把4放到了set,这意味着无论以后2类操作的r是多少,4都有可能成为答案. 然后对于(2,4,4), 我们去查(5,6)中的大于等于4的最小值,查出来是5,再查set里的大于等于4的最小值,查出来是4,所以答案是4.

对于(1,5),把5放到set里,对于(1,1,5),我们查询{2,4,3,5,1}中大于等于5的最小值,和set{4,5}中的大于等于5的最小值,这就是为什么我们不需要把5加上10000000,因为5一定存到set里了,如果答案是5,主席树的查询和set的查询肯定都会是5, 如果答案不是5,那5更不会有什么影响了

 

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#define mem(a,x) memset(a,x,sizeof(a))
#define gi(x) scanf("%d",&x)
#define gi2(x,y) scanf("%d%d",&x,&y)
#define gi3(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define gll(x) scanf("%lld",&x)
#define gll2(x,y) scanf("%lld%lld",&x,&y)
using namespace std;
const double eps=1e-8; 
typedef long long ll;
const int MAXN=100005;
const ll mod=1e7+9;
const int inf=0x3f3f3f3f;
struct Node{
	int L,R;
	int sum;
}node[MAXN*20];
int a[MAXN];
int root[MAXN];
int n,m,cnt;
int book[MAXN];
sets;

int read()     //输入外挂
{
	int res=0,ch,flag=0;
	if((ch=getchar())=='-')
		flag=1;
	else if(ch>='0'&&ch<='9')
		res=ch-'0';
	while((ch=getchar())>='0'&&ch<='9')
		res=res*10+ch-'0';
	return flag?-res:res;
}
void Out(int a)    //输出外挂
{
	if(a>9)
		Out(a/10);
	putchar(a%10+'0');
}
void init(){
	for(int i=1;i<=n;i++){
		book[i]=0;
	}
	s.clear();
	root[0]=0;
	cnt=1;
	node[0].L=node[0].R=node[0].sum=0;
	for(int i=0;i>1;
	if(p<=mid)update(p, node[rt].L, l, mid);
	else update(p, node[rt].R, mid+1, r);
}
int query(int rt,int l,int r,int L,int R){
	if(R=L&&r<=R){
		if(l==r)return l;
		else{
			int mid=(l+r)>>1;
			int res=-1;
			if(L<=mid)res=query(node[rt].L, l, mid, L, R);
			if(R>=mid+1&&res==-1)res=query(node[rt].R, mid+1, r, L, R);
			return res;
		}
	}
	int mid=(l+r)>>1;
	int res=-1;
	if(L<=mid)res=query(node[rt].L, l, mid, L, R);
	if(R>=mid+1&&res==-1)res=query(node[rt].R, mid+1, r, L, R);
	return res;
}

int main(){
	int T;T=read();
	while(T--){
		n=read(),m=read();
		init();
		for(int i=1;i<=n;i++)a[i]=read();
		root[n+1]=0;
		for(int i=n;i>=1;i--){
			root[i]=root[i+1];
			update(a[i], root[i], 1, n);	
		}
		int lastans=0;
		for(int i=1;i<=m;i++){
			int id;
			gi(id);
			if(id==1){
				int t;t=read();
				int pos=t^lastans;
				if(book[pos])continue;
				book[pos]=1;
				s.insert(a[pos]);
			}
			else{
				int t1,t2;
				t1=read(),t2=read();
				int R=(t1^lastans)+1;
				int k=t2^lastans;
				int ans=n+1;
				set::iterator it =s.lower_bound(k);
				if(it!=s.end())ans=*it;
				if(R<=n){
					t1=query(root[R], 1, n, k, n);
					if(t1!=-1){
						ans=min(ans,t1);
					}
				}
				Out(ans);
				putchar('\n');
				lastans=ans;
			}
		}
	}
	return 0;
}

 

 

你可能感兴趣的:(主席树)