给你具有n个结点n−1条边的无向无环连通图,结点编号1∼n,每条边上有一个数作为他的边权,定义函数f(i,j)为连接i,j的简单路径的所有边权的异或值。
输入描述
输入的第一行为一个整数T,代表测试用例的组数
接下来的T组测试用例按照如下格式给出:
每组数据占n行,第一行有1个整数n,接下来的n−1行,每行有3个整数u,v,w分别表示每条边的起点、终点、权值
输出描述
对于每组测试数据,在新的一行中输出答案
样例输入
2
1
2
1 2 3
样例输出
0
3
求解公式的大致意思是算出所有f(i,j)的和,而对于每个边来说,它由两端点组成,设数组d[i](表示以1为根的树中以i为根的子树中的节点的个数),所以一条边出现的个数是:(n-d[i])*d[i],对于所有边遍历一边就行了。
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6+1;
int d[maxn], res;
struct edge{
int to, next, w; edge(){}
edge(int u, int v, int w): to(u), next(v), w(w){}
}e[maxn]; int head[maxn], tot;
void dfs1(int x, int pre) {
d[x] ++;
for(int i = head[x]; ~i; i = e[i].next) {
int j = e[i].to;
if (j == pre) continue;
dfs1(j, x);
d[x] += d[j];
}
}
void dfs2(int x, int pre) {
for(int i = head[x]; ~i; i = e[i].next) {
int j = e[i].to;
if (j == pre) continue;
int t = (d[1] - d[j]) * d[j];
if (t & 1) res ^= e[i].w;
dfs2(j, x);
}
}
int main() {
int T;
scanf("%d", &T);
while(T --) {
memset(head, -1, sizeof(head));
memset(d, 0, sizeof(d));
int n; tot = 0, res = 0;
scanf("%d", &n);
for(int i = 1; i < n; ++ i) {
int u, v, w;
scanf("%d%d%d", &u, &v, &w);
e[++tot] = edge(u, head[v], w), head[v] = tot;
e[++tot] = edge(v, head[u], w), head[u] = tot;
}
dfs1(1, 1);
dfs2(1, 1);
printf("%d\n", res);
}
return 0;
}