题目: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;
}