【AGC010 C】Cleaning

题意

　　有一棵 \(n\) 个点的树，第 \(i\) 个节点有 \(a_i\) 个石子。

　　每次都可以选择一对不同的叶子节点，这对叶子节点路径上的所有点都必须要有石子。然后去掉这两个叶子节点路径上的每个节点中的一颗石子（选择的两个叶子节点也是路径中的点）。叶子节点是度数为 \(1\) 的点。

　　判断是否有可能去掉所有的石头。

　　\(n\le 10^5\)

　　\(a_i\le 10^9\)

题解

　　https://blog.csdn.net/zjznku/article/details/54948774

#include<bits/stdc++.h>
#define ll long long
#define N 100002
using namespace std;
inline int read(){
	int x=0; bool f=1; char c=getchar();
	for(;!isdigit(c); c=getchar()) if(c=='-') f=0;
	for(; isdigit(c); c=getchar()) x=(x<<3)+(x<<1)+(c^'0');
	if(f) return x;
	return 0-x;
}
int n,a[N],du[N];
struct edge{int v,nxt;}e[N<<1];
int hd[N],cnt;
inline void add(int u, int v){e[++cnt]=(edge){v,hd[u]}, hd[u]=cnt, ++du[u];}
void nosol(){
    puts("NO");
    exit(0);
}
void dfs(int u, int fa){
    if(du[u]==1) return;
    ll sum=0; int mx=0;
    for(int i=hd[u]; i; i=e[i].nxt) if(e[i].v!=fa){
        dfs(e[i].v,u);
        sum+=a[e[i].v];
        mx=max(mx,a[e[i].v]);
    }
    ll x=sum-a[u], y=2*a[u]-sum;
    if(x<0 || y<0 || x>min(sum-mx,sum/2) || (!fa && y)) nosol();
    a[u]=y;
}
int main(){
    n=read();
    for(int i=1; i<=n; i++) a[i]=read();
    if(n==2){
        if(a[1]==a[2]) puts("YES");
        else puts("NO");
        return 0;
    }
    int u,v;
    for(int i=1; i<n; i++){
        u=read(), v=read();
        add(u,v), add(v,u);
    }
    int rt=1;
    while(du[rt]==1) ++rt;
    dfs(rt,0);
    if(a[rt]) puts("NO");
    else puts("YES");
    return 0;
}