Query on A Tree(2017ACM/ICPC广西邀请赛)

类似可持久化线段树,这树不是维护区间了,是维护32位中0和1的个数,l代表1,r代表0,建树,利用前缀和思想,找出节点v~u中的01字典树,选择最大异或的那个数就可以了。

#include <iostream>
#include <unordered_map>
#include <vector>
#include <deque>
#include <algorithm>
#include <map>
#include <set>
#include <cstring>
#include <cmath>
#include <queue>
#include <stack>
#include <iomanip>
//#define int  long long
#define lc(x) tr[x].l
#define rc(x) tr[x].r

using namespace std;
typedef  long long ll;
typedef pair<int, int> PII;

const int N = 2e5 + 10;
int n, m;
int h[N], ne[N],w[N], e[N], idx,child_count[N],root[N],cnt_1;
int id[N], cnt;
struct node{
	int l=0,r=0,cnt=0;
}tr[N*22];//N*(log(N)+3);

void add(int a,int b){
	e[idx]=b;
	ne[idx]=h[a];
	h[a]=idx++;
}

void init(){
	memset(h,-1,sizeof h);
	for(int i=0;i<=cnt_1;i++){
		tr[i].l=tr[i].r=tr[i].cnt=0;
		root[i]=0;
	}
	memset(child_count,0,sizeof child_count);
	memset(id,0,sizeof id);
	idx=0,cnt=0,cnt_1=0;
}

void insert(int x,int &y,int value){//遵循左1右0的规则;
	y=++cnt_1,tr[y]=tr[x];
	int now=y,pre=x;
	for(int i=31;i>=0;i--){//循环建树
		int t=(value>>i)&1;
		if(t==1){
			tr[now].l=++cnt_1;
			now=tr[now].l;
			tr[now]=tr[tr[pre].l];
			pre=tr[pre].l;
			tr[now].cnt++;
		}
		else{
			tr[now].r=++cnt_1;
			now=tr[now].r;
			tr[now]=tr[tr[pre].r];
			pre=tr[pre].r;
			tr[now].cnt++;
		}
//		cout<<"pre = "<<tr[pre].cnt<<" now = "<<tr[now].cnt<<" w = "<<((value>>i)&1)<<" cnt_1 = "<<cnt_1<<endl;
	}
}

int query(int x,int pre,int now){
	int res=0;
//	cout<<"Pre = "<<pre<<" now = "<<now<<"\n";
	for(int i=31;i>=0;i--){
		int t=(x>>i)&1;
		if((t==0&&tr[tr[now].l].cnt>tr[tr[pre].l].cnt)||(t==1&&tr[tr[now].r].cnt>tr[tr[pre].r].cnt)){
			res=(res<<1)+(t^1);
			if(t==0){
				now=tr[now].l;
				pre=tr[pre].l;
			} 
			if(t==1){
				now=tr[now].r;
				pre=tr[pre].r;
			}
//			cout<<tr[tr[pre].r].cnt<<" "<<tr[tr[now].r].cnt<<"\n";
		}
		else{
			res=(res<<1)+t;
			if(t==0){
				now=tr[now].r;
				pre=tr[pre].r;
			}
			if(t==1) {
				now=tr[now].l;
				pre=tr[pre].l;
//				cout<<"++ ";
			}
		}
//		cout<<" pre = "<<tr[pre].cnt<<" tr[now].cnt = "<<tr[now].cnt<<" now = "<<now<<" "<<" t = "<<t<<endl;
	}
//	cout<<"res = "<<res<<endl;
	return res^x;
}

void dfs1(int u){
	id[u]=++cnt,child_count[u]=1;
	insert(root[id[u]-1],root[id[u]],w[u]);
//	cout<<"id = "<<id[u]<<" root = "<<root[id[u]]<<"\n";
	for(int i=h[u];~i;i=ne[i]){
		int v=e[i];
		dfs1(v);
		child_count[u]+=child_count[v];
	}
}

void solve() {
	int b;
	while (scanf("%d %d", &n, &m) != -1) {
		init();
		for(int i=1;i<=n;i++) scanf("%d",&w[i]);
		for(int i=2;i<=n;i++){
			scanf("%d",&b);
			add(b,i);
		}
		dfs1(1);//给节点从新编个号
		//cnt=0;
		int x,y;
		while(m--){
			scanf("%d %d",&y,&x);
			printf("%d\n",query(x,root[id[y]-1],root[id[y]+child_count[y]-1]));
		}
	}
}

signed main() {
	int t = 1;
	//cin>>t;
	while (t--) solve();
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值