NC20857 Xor Path
Xor Path
https://ac.nowcoder.com/acm/problem/20857
题目大意
定义 为树上编号为 两点路径上的点权异或和。求所有 的异或和。
题解
由于都是异或和,因此只需要对每一个点考虑其在所有路径中出现的次数即可。
将出现分成三个部分:点出现在其子树内点和外部点的路径上、点连接其子树内点的路径上、点出现在其子树内点之间的路径上。分别统计即可。
时间复杂度为 。可以一次 DFS 实现,我这里用了两次。
#include <bits/stdc++.h> #define REP(temp, init_val, end_val) for (int temp = init_val; temp <= end_val; ++temp) #define REPR(temp, init_val, end_val) for (int temp = init_val; temp >= end_val; --temp) using namespace std; int read(){ int f = 1, x = 0; char c = getchar(); while (c < '0' || c > '9'){if(c == '-') f = -f; c = getchar();} while (c >= '0' && c <= '9')x = x * 10 + c - '0', c = getchar(); return f * x; } int n, a[500005], siz[500005] = {0}, ans = 0; int to[1000005], nxt[1000005], at[500005] = {0}, cnt = 0; void dfs(int cur, int fa){ siz[cur] = 1; for (int i = at[cur]; i; i = nxt[i]){ int v = to[i]; if (v == fa) continue; dfs(v, cur); siz[cur] += siz[v]; } } void dfs2(int cur, int fa){ int sum = (n - siz[cur]) & 1, sum2 = 0; sum = (sum * siz[cur]) & 1; sum = (sum + (siz[cur] - 1)) & 1; for (int i = at[cur]; i; i = nxt[i]){ int v = to[i]; if (v == fa) continue; sum = (sum + siz[v] * sum2) & 1; sum2 = (sum2 + siz[v]) & 1; dfs2(v, cur); } if (sum & 1) ans ^= a[cur]; } void init(){ n = read(); REP(i, 1, n - 1){ int u = read(), v = read(); to[++cnt] = v, nxt[cnt] = at[u], at[u] = cnt; to[++cnt] = u, nxt[cnt] = at[v], at[v] = cnt; } REP(i, 1, n) a[i] = read(); } void solve(){ dfs(1, 0); dfs2(1, 0); printf("%d\n", ans); } int main(){ init(); solve(); return 0; }