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;
}
全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务