「BZOJ3727 PA2014」Final Zadanie 思维

题面

分析

不妨看看，$b[fa[x]]-b[x]$ 等于什么，手算一下，发现是 $x$ 子树的总人数 - 不为 $x$ 子树成员的总人数。（这个性质还是很容易发现的。）

然后我想的是（如果你只想看正解，后四排可跳过。）

令 $c[x]=b[fa[x]]-b[x]$。令 $x$ 为现在的结点，$son$ 为其儿子，$up$ 为它上面的一堆元素。

则 $\sum c[son]=-(sizeson-1)\times sumson-sizson\times (x+up)$

又 $c[x]=sumson+x-up$

你会发现，做到这里，有三个变量，两个方程，做不了。

考虑算出所有结点的人数和。

令 $sum$ 为所有节点的人数和，$zs[x]$ 为子树的人数和。

好，又看到这个式子：$b[fa[x]]-b[x]$ = $x$ 子树的总人数 - 不为 $x$ 子树成员的总人数。

我们不妨换个写法，用带有 $sum$ 的式子表达。

$b[fa[x]]-b[x]=2zs[x]-sum$

所以 $\sum b[fa[x]]-b[x]=\sum 2zs[x]-(n-1)\times sum$

ps:这里是核心，也是难点，妙的在于它巧妙地利用了原本的条件。

考虑把 $zs[x]$ 约去，这里有个很巧妙的处理，我也是看了题解才明白的。。。（我太菜了）

对于一个树上的一个节点 $x$，它对 $\sum zs[i]$ 的贡献为它到 $1$ 的距离，则 $\sum zs[i]$ 刚好为 $b[1]$。

所以 $\sum b[fa[x]]-b[x]=2\times b[1]-(n-1)\times sum(x!=1)$

这样就可以算出 $sum$。

算出这个后，我们就可以知道 $x$ 子树的总人数 + 不为 $x$ 子树成员的总人数，与 $b$ 数组联立，即可算出子树和。

$ans[x]=zs[x]-\sum zs[son[x]](x!=1)$。
$ans[1]=sum-\sum ans[i](i!=1)$

Code

#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <vector>
#define LL long long
using namespace std;
const int MAXN = 3e5 + 5;
int n, b[MAXN], a[MAXN];
LL sum_, ans[MAXN], zs[MAXN]; 
vector <int> v[MAXN];
void dfs(int x, int fa) {
	if(~fa) sum_ += b[fa] - b[x];
	for(unsigned int i = 0; i < v[x].size(); i ++) {
		int Y = v[x][i]; if(Y == fa) continue; dfs(Y, x);
	}
}
int dfs1(int x, int fa) {
	if(~fa) zs[x] = (b[fa] - b[x] + sum_) / 2;
	LL tmp = 0;
	for(unsigned int i = 0; i < v[x].size(); i ++) {
		int Y = v[x][i]; if(Y == fa) continue;
		tmp += dfs1(Y, x);
	}
	ans[x] = zs[x] - tmp;
	return zs[x];
}
int main() {
	int x, y;
	scanf("%d", &n);
	for(int i = 1; i < n; i ++) scanf("%d%d", &x, &y), v[x].push_back(y), v[y].push_back(x);
	for(int i = 1; i <= n; i ++) scanf("%d", &b[i]);
	dfs(1, -1); sum_ = (b[1] * 2 - sum_) / (n - 1); dfs1(1, -1); ans[1] = sum_;
	for(int i = 2; i <= n; i ++) ans[1] -= ans[i];
	for(int i = 1; i <= n; i ++) printf("%lld ", ans[i]);
	return 0;
}