题目大意:国家有N个城市,1号城市为首都,有M个国民,每个国民都在首都工作,晚上返回家中,给定每个城市有多少国民居住。每个城市都有一个心情检测器,当国民经过城市时,心情检测器根据国民的心情加减1,但是心情检测器并不精确,所以要求你去判断在所有国民回家之后心情检测器上的数值是否正确。
贴个代码吧,其中有一个求以某一个结点为根的所有子节点权值和的操作,记录一下。
另外,有个条件自己比赛的时候推错了,就是:
假定:当前经过这一点的人数为y(快乐与不快乐人数之和),而这一点的快乐减不快乐人数之差是x,我们设快乐的人数为a,不快乐的人数为b,并默认a>b。然后我们得到 a+b=y,a-b=x推出x+y=2*a,所以说明像符合题意必须满足x+y是偶数。(比如人数为7,但是7能凑出的人数是有限的,4就不行,所以必定存在一个判断条件)
AC代码:
#include <iostream>
#include <cstdio>
#include <vector>
#include <cmath>
#include <cstring>
#include <algorithm>
//#define int long long
using namespace std;
const int maxn=1e6+5;
int p[maxn],h[maxn],a[maxn],b[maxn];
bool flag=true;
vector<int>g[maxn];
void dfs(int x,int fa) {
a[x]=p[x];
int sum_b=0;
for(int i=0; i<g[x].size(); i++) {
int v=g[x][i];
if(v==fa)continue;
dfs(v,x);
a[x]+=a[v];//递归计算经过当前点的所有人数
sum_b+=b[v];//计算子节点开心人数的总人数
}
if((a[x]+h[x])%2)flag=false;
b[x]=(a[x]+h[x])/2;
if(b[x]<0||b[x]>a[x])flag=false;
if(sum_b>b[x])flag=false;
}
int main() {
int t;
cin>>t;
while(t--) {
int n,m;
cin>>n>>m;
for(int i=1; i<=n; i++)g[i].clear();
for(int i=1; i<=n; i++)cin>>p[i];
for(int i=1; i<=n; i++)cin>>h[i];
for(int i=1; i<n; i++) {
int x,y;
cin>>x>>y;
g[x].push_back(y);
g[y].push_back(x);
}
flag=true;
dfs(1,-1);
if(flag)cout<<"YES"<<endl;
else cout<<"NO"<<endl;
}
}