lpoj5576 hongrock的柠檬树

题目：https://lpoj.cn/problemdetail?problemID=5576

输入一棵树，求树上所有两点之间路径或之和。

权神设计出来是用并查集做的。

然而第一反应就是向上合并维护二进制位。写了一发维护1的个数愉快的wa了，之后就忘了有这题了。最近这题被大师兄秒掉了，被教育了一手可以先当成整棵树全是1来做，然后再合并0的个数，减掉0的情况。

瞬间好写了很多。。。。

我们把子树0的个数向上合并的时候，检查一下父节点这一位是否是0，如果是0才合并，因为父节点是所以子节点的必经之路，如果父节点是1，那么子节点经过父节点往上或出来这一位就是1了。

ac代码：

#include<bits/stdc++.h>

using namespace std;

const int maxn = 1e5 + 5;
typedef long long ll;

ll ans;
int n, val[maxn];
int p[20], sz[maxn][20];
vector<int> G[maxn];


void crack(int x) {
    for (int i = 0; i < 20; ++i) {
        sz[x][i] = !(val[x] >> i & 1);
    }
}

void dfs(int x, int fa) {
    crack(x);

    for (auto v:G[x]) {
        if (v == fa) {
            continue;
        }
        dfs(v, x);
        for (int i = 0; i < 20; ++i) {
            ans -= 1LL * p[i] * sz[x][i] * sz[v][i];
            if (!(val[x] >> i & 1)) {
                sz[x][i] += sz[v][i];
            }
        }
    }
}

int main() {
    ios::sync_with_stdio(0);
    cin >> n;
    for (int i = 1; i <= n; ++i) {
        cin >> val[i];
    }

    int u, v;
    for (int i = 1; i < n; ++i) {
        cin >> u >> v;
        G[u].push_back(v);
        G[v].push_back(u);
    }

    p[0] = 1;
    for (int i = 1; i < 20; ++i) {
        p[i] = p[i - 1] << 1;
    }

    for (int i = 0; i < 20; ++i) {
        ans += 1LL * p[i] * n * (n - 1) / 2;
    }

    dfs(1, 0);
    cout << ans;
    return 0;
}