2021牛客暑期多校训练营4 Tree Xor

2021牛客暑期多校训练营4

Tree Xor

题目链接

题意

在一棵树上，对于每一个节点都有一个权值$w_{1...n}$，且权值的范围为$l_i\le w_i\le r_i$。

给定的值是每一个节点的权值范围$[l_i,r_i]$，以及每一条边$(u,v)$的$w_u\oplus w_v$的值。

求有多少种节点权值的赋值策略。

思路

对于节点$u$和节点$v$，假设$w_u\oplus w_v$的值为x，如果节点$u$的权值为$w_u=0$，那么节点$v$的权值就唯一确定$w_v=(w_u\oplus w_v)\oplus w_u=x$。

如果$w_u=0$异或上一个值$a$，此时$w_u=a$，且$w_v$变为$a\oplus x$。

所以，$[l_v,r_v]$​异或x后得到若干个小区间$S$，将$S$与$[l_u,r_u]$​取交集，先dfs到叶子节点，然后不断向上合并。

因为区间$[l_i,r_i]$异或上一个值x后得到的不是一段连续的区间，不好处理。

所以这里用到了线段树，对每一个节点的区间建立一个线段树，这里先用例子说一下线段树的性质。

可以发现，对于每一个节点$[l_i,r_i]$​，$l_i$​二进制位的高几位和$r_i$​的相同，并且剩下低位在$l_i$中是全0，在$r_i$中是全1。

那么，如果区间$[l_i,r_i]$异或上x后得到的区间为$[r_i\oplus x,l_i\oplus x]$​。

建好线段树后，只需要向上合并就行了。

合并的一些细节在代码的注释里面。

AC的代码

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
const int MAX = (1 << 30) - 1;
int n, l[N], r[N], tot = 0, rt[N];
vector<int> vec[N], val[N];

struct Node {
	int ls, rs;
	int	v; // v是标记
	// 如果v=2表示该节点存在左右子树(需要继续向下遍历)
	// 如果v=1表示该节点的区间存在
	// 如果v=0表示该节点的区间不存在

}t[25000010];


void pushdown(int i) {
	// 下传标记，并且动态开点（防止炸空间）
	if (!t[i].ls) t[i].ls = ++tot;
	if (!t[i].rs) t[i].rs = ++tot;
	if (t[i].v != 2) {
		t[t[i].ls].v = t[i].v;
		t[t[i].rs].v = t[i].v;
	}
	t[i].v = 2;
}


// [ql,qr]表示存在的区间
void build(int i, int l, int r, int ql, int qr) {
	if (l == ql && r == qr) {
		t[i].v = 1;
		return;
	}
	pushdown(i);
	/*
	分成3种情况：
		1. 存在的区间都在左子树；
		2. 存在的区间都在右子树；
		3. 两棵子树都有。
	*/
	int mid = (l + r) >> 1;
	if (qr <= mid) {
		build(t[i].ls, l, mid, ql, qr);
	}
	else if (ql > mid) {
		build(t[i].rs, mid + 1, r, ql, qr);
	}
	else {
		build(t[i].ls, l, mid, ql, mid);
		build(t[i].rs, mid + 1, r, mid + 1, qr);
	}
}

void exchange(int i, int l, int r, int v, int d) {
	if (l == r)
		return;
	if ((v >> d) & 1) {
		swap(t[i].ls, t[i].rs);
	}
	int mid = (l + r) >> 1;
	if (t[i].ls) {
		exchange(t[i].ls, l, mid, v, d - 1);
		exchange(t[i].rs, mid + 1, r, v, d - 1);
	}
}



void merge(int i, int I, int l, int r, int v, int d) {
	if (l == r) {
		t[i].v &= t[I].v;// 到叶子节点时，v的值为1或0，只有两棵树的这个点都存在时，才能为1
		return;
	}
	if((v >> d) & 1) {
		// v为相邻的两个节点的异或值
		// 如果v二进制位的这一位为1，区间异或后，左右子树需要交换
		swap(t[I].ls, t[I].rs);
	}
	int mid = (l + r) >> 1;
	if (t[i].v == 2) {
		if (t[I].v == 2) {
			pushdown(i); pushdown(I);
			merge(t[i].ls, t[I].ls, l, mid, v, d - 1);
			merge(t[i].rs, t[I].rs, mid + 1, r, v, d - 1);
		}
		else {
			if (t[I].v == 0)
				t[i].v = 0;
		}
	}
	else {
		if (t[I].ls) {
			if (t[i].v == 0) {
				return;
			}
			t[i].v = t[I].v; t[i].ls = t[I].ls; t[i].rs = t[I].rs;
			exchange(t[i].ls, l, mid, v, d - 1);
			exchange(t[i].rs, mid + 1, r, v, d - 1);
		}
		else {
			t[i].v &= t[I].v;
		}
	}

}

void dfs(int x, int fa) {
	rt[x] = ++tot;// rt表示每一个线段树的根节点
	build(rt[x], 0, MAX, l[x], r[x]);
	for (int i = 0; i < vec[x].size(); i++) {
		if (vec[x][i] != fa) {
			dfs(vec[x][i], x);
			merge(rt[x], rt[vec[x][i]], 0, MAX, val[x][i], 29);// 向上合并
		}
	}

}

int sum(int i, int l, int r) {
	if (t[i].v != 2) {
		if (t[i].v == 1) {
			return r - l + 1;
		}
		else {
			return 0;
		}
	}
	int mid = (l + r) >> 1;
	return sum(t[i].ls, l, mid) + sum(t[i].rs, mid + 1, r);
}

int main() {
	cin >> n;
	for (int i = 1; i <= n; i++)	cin >> l[i] >> r[i];
	for (int i = 1; i < n; i++) {
		int u, v, w;
		cin >> u >> v >> w;
		vec[u].push_back(v); val[u].push_back(w);
		vec[v].push_back(u); val[v].push_back(w);// 存储无向图
	}
	dfs(1, -1);
	cout << sum(1, 0, MAX);
	return 0;
}