标签：Partitioning Bakry puts NO 点权 CodeForces else ll getchar

链接：

洛谷

题目大意：

一棵树有 \(n\) 个节点，第 \(i\) 个节点的点权为 \(a_i\)。

你需要回答：能不能选择这棵树中的至少 \(1\) 条边、至多 \(k-1\) 条边删除，使得删除完这些边的树每个联通块的点权异或和相等。

思路：

\(a\oplus a=0\) 真的好用，就可以直接搜索了。

代码：

const int N = 1e5 + 10;

inline ll Read() {
	ll x = 0, f = 1;
	char c = getchar();
	while (c != '-' && (c < '0' || c > '9')) c = getchar();
	if (c == '-') f = -f, c = getchar();
	while (c >= '0' && c <= '9') x = (x << 3) + (x << 1) + c - '0', c = getchar();
	return x * f;
}

int t, n, k;
int a[N];
int head[N], tot;
struct edge {
	int to, nxt;
}e[N << 1];
void Add(int u, int v) {
	e[++tot] = (edge) {v, head[u]}, head[u] = tot; 
}

int cnt, xsum;
int dfs(int u, int fa) {
	int ans = a[u];
	for (int i = head[u]; i; i = e[i].nxt) {
		int v = e[i].to;
		if (v == fa) continue;
		int tmp = dfs(v, u);
		if (tmp == xsum) cnt++;
		else ans ^= tmp;
	}
	return ans;
}

int main() {
//	freopen(".in", "r", stdin);
//	freopen(".out", "w", stdout);
	for (t = Read(); t--; ) {
		n = Read(), k = Read();
		xsum = cnt = tot = 0;
		for (int i = 1; i <= n; i++) xsum ^= a[i] = Read();
		memset (head, 0, sizeof head);
		for (int i = 1; i < n; i++) {
			int u = Read(), v = Read();
			Add(u, v), Add(v, u);
		}
		if (!xsum) puts("YES");
		else {
			if (k >= 3) {
				dfs (1, 0);
				if (cnt >= 2) puts("YES");
				else puts("NO");
			} else puts("NO");
		}
	}
	return 0;
}

标签：Partitioning,Bakry,puts,NO,点权,CodeForces,else,ll,getchar
来源： https://www.cnblogs.com/GJY-JURUO/p/15576345.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。