【CF1388C】Uncle Bogdan and Country Happiness（树形dp、贪心）

快乐检测器

题意：

思路：

• 抓住重要的点，心情只有可能由好变坏，一个人变成坏心情后就不能变回好心情，但好心情却可以随时通过一条边变成坏心情。

• 这就导致我们可以狠狠地贪心，尽可能让多的人保持好心情，即使好心情多了也可以通过走边减少。

• 所以，对于每个点维护一个至少的好心情人数（因为以下结点为了满足检测值，需要一定的好心情人），维护一个走到该点的人数，然后再处理细节问题。

$Code:$

#include<bits/stdc++.h>
#include<unordered_map>
#define mem(a,b) memset(a,b,sizeof a)
#define cinios (ios::sync_with_stdio(false),cin.tie(0),cout.tie(0))
#define sca scanf
#define pri printf
#define forr(a,b,c) for(int a=b;a<=c;a++)
#define rfor(a,b,c) for(int a=b;a>=c;a--)
#define all(a) a.begin(),a.end()
#define oper(a) (operator<(const a& ee)const)
#define endl "\n"
using namespace std;

typedef long long ll;
typedef pair<int, int> PII;

double DNF = 1e17;
const int N = 100010, M = 200010, MM = 110;
int INF = 0x3f3f3f3f, mod = 998244353;
ll LNF = 0x3f3f3f3f3f3f3f3f;
int n, m, k, T, S, D, K;
int h[N], ne[M], e[M], idx;
int a[N], p[N], good[N];

void add(int a, int b) {
	e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

bool dfs(int x, int fa) {
	// good 记录该点往下已经确认了的好心情的人数（包括该点）
	// p 维护这个点经过的人数
	// a 记录的是每个结点的检测值

	int cnt = 0;
	for (int i = h[x]; ~i; i = ne[i]) {
		int j = e[i];
		if (j == fa)continue;
		if (!dfs(j, x))return false;
		cnt += good[j];
		p[x] += p[j];
	}

	a[x] -= cnt, p[x] -= cnt;
	//已经确定的好心情人肯定可以去掉，对结果（公式）不影响
	//去掉后，p 记录的就是经过该节点时，心情还不确定的人数
	// 
	//设好心情人数为 a，坏心情为 b
	//有公式：a + b = p[x]，a - b = a[x]

	//显然要保证以下俩情况不成立
	if (abs(a[x]) > abs(p[x]) || (a[x] + p[x]) % 2)return false;

	good[x] = (p[x] + a[x]) / 2;
	good[x] += cnt;
	p[x] += cnt;//不要忘记回溯 p

	return true;
}

void solve() {
	cin >> n >> m;
	forr(i, 1, n)cin >> p[i], good[i] = 0;
	forr(i, 1, n)cin >> a[i];

	mem(h, -1), idx = 0;
	forr(i, 1, n - 1) {
		int a, b;
		cin >> a >> b;
		add(a, b), add(b, a);
	}

	if (!dfs(1, -1))cout << "NO\n";
	else cout << "YES\n";
}

int main() {
	cinios;
	cin >> T;
	while (T--)solve();
	return 0;
}
/*
*/