[BZOJ]3727 Final Zadanie 公式推导

3727: PA2014 Final Zadanie

Time Limit: 10 Sec   Memory Limit: 128 MB
Submit: 319   Solved: 138
[ Submit][ Status][ Discuss]

Description

吉丽YY了一道神题，题面是这样的：
“一棵n个点的树，每条边长度为1，第i个结点居住着a[i]个人。假设在i结点举行会议，所有人都从原住址沿着最短路径来到i结点，行走的总路程为b[i]。输出所有b[i]。”
吉丽已经造好了数据，但熊孩子把输入文件中所有a[i]给删掉了。你能帮他恢复吗？

Input

第一行一个整数n(2<=n<=300000)。
接下来n-1行，每行两个整数x,y，表示x和y之间有连边。
接下来一行由空格隔开的n个整数b[i](0<=b[i]<=10^9)。

Output

输出一行由空格隔开的n个整数a[i]。
如果你觉得有多组解就任意输出其中一组。

Sample Input

2
1 2
17 31

Sample Output

31 17

HINT

Source

鸣谢Jcvb

[ Submit][ Status][ Discuss] 

HOME Back

达哥竟然出的原题... 假设一个sum[1]推导出所有的a， 再计算与真实的sum[1]之间的差值就可以了.

好吧有点模糊... 自己写的详解具体在这套的第二题 : Click Here 

班上人看见我这个假设法挺懵逼的...尤其是看见第35行的随便假设.... 大胆猜想， 理性推测.

#include<stdio.h>
#include<cstring>
#define Mercer   register int
#define clear(a) memset(a, 0, sizeof(a))
const int maxn = 300005;
typedef long long dnt;
dnt all, punchh, delta;
int num, n, t, T, fake, rightt;
int h[maxn], a[maxn], b[maxn], sum[maxn], dis[maxn], cnt[maxn];
inline const int read(){
	register int x = 0;
	register char ch = getchar();
	while(ch < '0' || ch > '9') ch = getchar();
	while(ch >= '0' && ch <= '9') x = (x << 3) + (x << 1) + ch - '0', ch = getchar();
	return x;
}
struct edge{ int nxt, v;}e[maxn * 2];
inline void add(int u, int v){
	e[++num].v = v, e[num].nxt = h[u], h[u] = num;
	e[++num].v = u, e[num].nxt = h[v], h[v] = num;
}
void dfss(int u, int fa){
	a[u] = sum[u];
	for(int i = h[u]; i; i = e[i].nxt){
		int v = e[i].v;
		if(v == fa) continue;
		sum[v] = (sum[1] - b[v] + b[u]) / 2;
		dis[v] = dis[u] + 1, dfss(v, u);
		a[u] -= sum[v];
		cnt[u]++;
	}
}
inline void work(){
	for(Mercer i = 1; i <= n; ++i) b[i] = read();
	if((10000000 + b[1] - b[fake]) & 1) sum[1] = 10000001;
	else sum[1] = 10000000;
	dfss(1, 1);
	for(Mercer i = 2; i <= n; ++i) all += (dnt) a[i] * dis[i], punchh += dis[i];
	delta = (b[1] - all) * 2;
	for(Mercer i = 2; i <= n; ++i) rightt += (dnt) dis[i] * (1 - cnt[i]);
	delta /= rightt, sum[1] += delta;
	dfss(1, 1);
}
int main(){
	n = read();
	for(Mercer i = 1; i < n; ++i){
		int u = read(), v = read();
		add(u, v);
		if(u == 1) fake = v;
		if(v == 1) fake = u;
	}
	work();
	for(Mercer i = 1; i < n; ++i) printf("%d ", a[i]);
	printf("%d\n", a[n]);
}