ICPC NEAU Programming Contest 2020~~M. 再来异或

给你具有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;
}