Problem - C - Codeforces
波格丹叔叔在弗林特船长的团队里待了很长一段时间,有时会怀念他的家乡。今天他告诉你他的国家是如何引入幸福指数的。有n个城市和n -1条连接城市的无方向道路。任何城市的公民都可以通过这些道路到达任何其他城市。城市编号从1到n,城市1是首都。换句话说,这个国家有一个树形结构。有100万公民居住在这个国家。有很多人住在第i个城市,但他们都在首都工作。傍晚时分,所有市民都选择最短路径返回各自的城市。每个人都有自己的心情:有的人带着好心情离开,有的人已经心情不好了。此外,任何人都可以破坏他的心情在回家的路上。如果一个人心情不好,他不会改善它。每个城市都安装了幸福探测器,以监测每个到访城市的人的幸福程度。第i个城市的检测器计算出幸福指数h,即心情好的人数减去心情不好的人数。简单地说,一个人的情绪在城市里是不变的幸福检测仪仍在开发中,所以判断一个人的幸福有可能会出错。一天深夜,当所有市民都顺利回家后,政府请Bogdan叔叔(全国最好的程序员)检查收集到的幸福指数是否正确。Bogdan叔叔成功地解决了这个问题。你也能做到吗?更正式地说,你需要检查:“有没有可能,在所有人都回家后,每个城市i的幸福指数都恰好等于hi”。输入第一行包含一个整数t (1 < t < 10000)——测试用例的数量。每个测试用例的第一行包含两个整数n和m (1 < n < 105;0 < m < 10°)-城市和公民的数量。每个测试用例的第二行包含n个整数p1,P2,,Pn(0≤pi < m;P1 + p2 +。+ pn = m),其中p;是居住在第i个城市的人数。第三行包含n个整数h1, h2,, hn(-10°< h;< 10°),其中h;计算出的幸福指数是第i个城市吗接下来的n - 1行包含道路的描述,每行一个。每行包含两个整数x;和yi (1 S xi, yi S n;和y;是由第i条公路连接的城市吗?可以保证所有测试用例的n的和不超过2 - 105。输出对于每个测试用例,如果收集的数据正确,则打印YES,否则打印NO。在任何情况下,都可以打印YES或NO中的字符。
Examples
input
Copy
2 7 4 1 0 1 1 0 1 0 4 0 0 -1 0 -1 0 1 2 1 3 1 4 3 5 3 6 3 7 5 11 1 2 5 2 1 -11 -2 -6 -2 -1 1 2 1 3 1 4 3 5
output
Copy
YES YES
input
Copy
2 4 4 1 1 1 1 4 1 -3 -1 1 2 1 3 1 4 3 13 3 3 7 13 1 4 1 2 1 3
output
Copy
NO NO
题解:
首先我们应该明白如果从根节点开始考虑的话,不确定的状态太多,不好考虑,所以应该从叶子节点考虑
对于题意来说不满足的情况有这么几种,
1.(子树城市居民数 + 城市心情值)% 2 == 1 (假设居民心情全为正的或全为负的,一个人心情变化,变化结果就是2)
2.假如说y是x的子节点,那么y的市民好心情数,肯定不能大于x(因为居民回家途中心情只可能变坏,不可能变好)
3.一个城市居民好心情数目,应该 = (子树城市居民数 + 城市心情值)/2 ,人数不能小于0
4,同理坏心情数目也不能小于0
剩下就是很完美的代码实现,非常简洁(只能说大佬的代码能力和常人确实不是一个级别)
#include <cstdio>
#include <cstring>
#include <algorithm>
#include<iostream>
#include<vector>
#include<set>
#include<map>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
#define int long long
vector<int> p[100050];
int good[100050];
int son[100050];
int h[100050];
int a[100050];
int f;
void dfs(int x,int fa)
{
son[x] = a[x];
int s = 0;
for(auto ne:p[x])
{
if(ne == fa)
continue;
dfs(ne,x);
son[x] += son[ne];
s += good[ne];
}
good[x] = (son[x] + h[x])/2;
if(abs(son[x] + h[x])%2 == 1||s > good[x] || good[x] < 0||son[x] - good[x] < 0)
{
f = 1;
}
}
void solve()
{
f = 0;
int n,m;
cin >> n >> m;
for(int i = 1;i <= n;i++)
{
cin >> a[i];
}
for(int i = 1;i <= n;i++)
{
cin >> h[i];
}
for(int i = 1;i < n;i++)
{
int l,r;
cin >> l >> r;
p[l].push_back(r);
p[r].push_back(l);
}
dfs(1,1);
if(f)
{
cout <<"NO\n";
}
else
{
cout <<"YES\n";
}
for(int i = 1;i <= n;i++)
{
p[i].clear();
son[i] = 0;
good[i] = 0;
}
}
signed main()
{
// ios::sync_with_stdio(0);
// cin.tie(0);cout.tie(0);
int t = 1;
cin >> t;
while(t--)
{
solve();
}
}