Codeforces Round #660 (Div. 2) C - Uncle Bogdan and Country Happiness

题意： 一共$n$个城市，$m$个上班族。城市之间的联通图是一棵树。给定每个城市的总人数$p_i$和幸福指数$h_i$，每个人都在首都$1$号点上班，每个人下班时的心情也不尽相同，在下班回家的路上，每个人都可能在回家路上的任意一个地点时，心情从开心变成不开心，但是不可能从不开心变成开心。问是否存在一种情况使得每个城市的幸福指数都与给定的$h_i$相同。其中$h_i$等于下班路上经过该点时开心的人数减去下班路上经过该点时不开心的人数。
数据范围：$1\le n\le 10^5,0\le m\le 10^9,\stackrel{n}{\sum _{i=1}}{p}_i=m,-10^9\le h_i\le 10^9$

题解：
对于每个城市，设下班路上经过该城市的人数为$siz[u]$，其中开心的人数为$good[u]$，不开心的人数是$bad[u]$

• $good[u]$是整数，$bad[u]$是整数
• $good[u]\ge 0,bad[u]\ge 0$
• $\sum good[uson]\le good[u]$

代码：

#include
using namespace std;

typedef long long ll;

template<typename T>
inline T Read(){
    T s = 0, f = 1; char ch = getchar();
    while(!isdigit(ch)) {if(ch == '-') f = -1; ch = getchar();}
    while(isdigit(ch)) {s = (s << 3) + (s << 1) + ch - '0'; ch = getchar();}
    return s * f;
}

#define read() Read()
#define readl() Read()

const int N = 1e5 + 10;
const int M = N << 1;

ll p[N], ha[N];
bool ans;
int h[N], e[M], ne[M], idx;
void add(int a, int b) {
	e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

ll siz[N];
ll good[N];

void dfs(int u, int fa) {
	siz[u] = p[u];
	for(int i = h[u]; i != -1; i = ne[i]) {
		int v = e[i];
		if(v == fa) continue;
		dfs(v, u);
		siz[u] += siz[v];
	} 
	good[u] = siz[u] + ha[u];
	if(good[u] & 1) ans = false; //小数 
	
	good[u] >>= 1;
	if(good[u] < 0 || good[u] > siz[u]) ans = false; //两者均不为负数 
	 
	ll sgood = 0;
	for(int i = h[u]; i != -1; i = ne[i]) {
		int v = e[i];
		if(v == fa) continue;
		sgood += good[v];
	}
	if(good[u] < sgood) ans = false; //good/bad都不小于子树 
}

void solve() {
	idx = 0;
	int n = read(), m = read();	
	for(int i = 1; i <= n; ++i) h[i] = -1;
	for(int i = 1; i <= n; ++i) p[i] = readl();
	for(int i = 1; i <= n; ++i) ha[i] = readl();
	for(int i = 1; i < n; ++i) {
		int a = read(), b = read();
		add(a, b), add(b, a);
	}
	
	ans = true;
	dfs(1, 0);
	puts(ans ? "YES" : "NO");
}

int main()
{
	int T = read();
	for(int i = 1; i <= T; ++i) {
		solve();
	}
}