Codeforces Round #660 (Div. 2) 参与排名人数13591
[codeforces 1388C] Uncle Bogdan and Country Happiness 树的遍历+树的子树节点数量的变形
总目录详见https://blog.csdn.net/mrcrack/article/details/103564004
在线测评地址https://codeforces.com/contest/1388/problem/C
Problem | Lang | Verdict | Time | Memory |
---|---|---|---|---|
C - Uncle Bogdan and Country Happiness | GNU C++17 | Accepted | 140 ms | 12500 KB |
该题开创了先河:第一次独立运用树的性质,解决树的问题,原来对树的问题,一团浆糊,慢慢有了感觉。
题目大意:n个人都从首都(城市1)出发,以最短路径回到归属的城市,每个人经过一个城市,好心情可以还是好心情,好心情也可以变成坏心情,但是坏心情不能变成好心情,坏心情只能一直是坏心情了。给出经过各个城市的人员的心情统计,问这种情况是否能发生,若能,输出YES,若不能,输出NO.
基本思路:吃透样例是关键。
经过该城市的总人数 可以通过 树的子树节点数量的变形 进行计算。
Input:
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:
NO
NO
Input:
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:
YES
YES
AC代码如下:
#include <stdio.h>
#define maxn 200010
int h[maxn],head[maxn],tot,num1[maxn],num2[maxn],num3[maxn],flag,n,m,num4[maxn],f[maxn];
struct node{
int to,next;
}e[maxn<<1];
void add_edge(int u,int v){//临界表
tot++,e[tot].to=v,e[tot].next=head[u],head[u]=tot;
}
void dfs1(int u,int fa){//计算经过每个城市的人数,确认城市间的父子关系
int v,b;
f[u]=fa;
for(b=head[u];b;b=e[b].next){
v=e[b].to;
if(v==fa)continue;//若是父节点,跳过
dfs1(v,u);
num1[u]+=num1[v];//num1[u]表示经过城市u的所有人数
}
}
void count(){
int a,u,b,v;
for(u=1;u<=n;u++){
a=num1[u]+h[u];
if(a%2){flag=1;return;}//a若是奇数,无效
num2[u]=a/2;//num2[u]表示经过城市u的好心情人数(包括城市u)
num3[u]=num1[u]-num2[u];//num3[u]表示经过城市u的坏心情人数(包括城市u)
if(num2[u]<0||num3[u]<0){flag=1;return;}//负数,无效
}
for(u=1;u<=n;u++)
for(b=head[u];b;b=e[b].next){
v=e[b].to;
if(v!=f[u])num4[u]+=num2[v];//num4[u]表示以城市u为父节点,直接相连的字节点城市的好心情人数(不包括城市u)
}
for(u=1;u<=n;u++)
if(num2[u]<num4[u]){flag=1;return;}//父节点的好心情人数 小于 直接相连的字节点城市的好心情人数,无效。
}
int main(){
int t,i,j,u,v;
scanf("%d",&t);
while(t--){
scanf("%d%d",&n,&m);
tot=0,flag=0;
for(i=1;i<=n;i++)head[i]=0,num4[i]=0;//初始化
for(i=1;i<=n;i++)scanf("%d",&num1[i]);
for(i=1;i<=n;i++)scanf("%d",&h[i]);
for(i=1;i<n;i++){
scanf("%d%d",&u,&v);
add_edge(u,v),add_edge(v,u);//无向图
}
dfs1(1,0);//计算经过每个城市的人数,确认城市间的父子关系
count();//计算num2[],num3[],num4[]
if(flag)printf("NO\n");
else printf("YES\n");
}
return 0;
}