有一棵N个节点的树,节点标号1-N,第i个边连着两个节点ai, bi。
每个节点有Ai个石头。每次你可以选择一对叶子节点(叶子节点是度数为1的点),然后将连接着两个节点的路径上的所有节点的石头数减一,
如果这条路径上存在石头数为0的点,那么不能进行这个操作。
问是否能把树上的所有石头全部移除。
数据
(2 <= N <= 1e5), (1 <= ai, bi <= N), (0 <= Ai <= 1e9), 数据保证给出一棵树。
输入
5
1 2 1 1 2
2 4
5 2
3 2
1 3
3
1 2 1
1 2
2 3
6
3 2 2 2 2 2
1 2
2 3
1 4
1 5
4 6
输出
YES
NO
YES
说明
对于样例1,选节点4,5,这样这条路径上除了4外都只有一个石头,4没有石头。
然后选择1,5,就完成了目标

解题方法:
题解描述来自: OTZ
主要是,我很难描述清楚,就借用一下博主的了, 写得非常清晰
首先可以将路径进行归类,对于一个非叶子节点A,经过其的路径可以分为两种情况:

1.经过一个子节点;

2.经过两个子节点。

路径经过两个子节点意味着这条路不用往上走了,即不会影响A的父亲。

路径经过一个子节点意味着需要再向上传递,因为另一个端点不在以A为根的子树里。

接着构思dfs的细节:

1.考虑叶子节点,叶子节点的权值一定得向上传递,因为叶子节点是路径的端点。

2.考虑非叶子节点A‘,已知权值是Ai,设从其儿子节点需要传上来的权值和是tot,其中第一种边贡献了x1,第二种边贡献了x2,其中,x1需要向上传递。容易列出方程:

x1+x2=Ai,2*x2+x1=tot,可以解出x1与x2的值。

接着要做的就是判断x1与x2的合法性:

1.x1与x2均要大于等于0;

2.当x1与x2均大于等于0的时候,要能真正画出这样的情况。观察发现,仅当儿子向上传递的权值分配很畸形的时候才画不出来。设maxw为儿子中向上传递权值的最大值,当maxw>Ai的时候,就不能画出x1条第一种路径,x2条第二种路径。动手模拟一下就可以看出来。

最后,看一下整棵树的根节点是不是x1等于0,若是的话,证明找到了可行解,因为经过根节点的只能是第二种边。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5+10;
const int maxm = 2e5+10;
typedef long long LL;
LL A[maxn]; int n, edgecnt, head[maxn];
struct edge{int v, nxt; } E[maxm];
void init(){
    edgecnt = 0; memset(head, -1, sizeof(head));
}
void addedge(int u, int v){
    E[edgecnt].v = v, E[edgecnt].nxt = head[u], head[u] = edgecnt++;
}
LL dfs(int u, int fa){
    int cnt = 0;
    for(int i = head[u]; ~i; i = E[i].nxt){
        if(u != E[i].v) cnt++;
    }
    if(cnt == 1){
        return A[u];
    }
    LL tot = 0, x1 = 0, x2 = 0, maxw = 0, sumw;
    for(int i = head[u]; ~i; i = E[i].nxt){
        int v = E[i].v;
        if(v == fa) continue;
        sumw = dfs(v, u);
        maxw = max(maxw, sumw);
        tot += sumw;
    }
    //x1 = 2 * A[u] - tot;
    x2 = tot - A[u];
    x1 = A[u]  -x2;
    if(x1 < 0 || x2 < 0 || A[u] < maxw || (x1 > 0 && fa == -1)){
        puts("NO");
        exit(0);
    }
    return x1;
}
int main(){
    init();
    scanf("%d", &n);
    for(int i = 1; i <= n; i++){
        scanf("%lld", &A[i]);
    }
    if(n == 1){
        if(A[1]) printf("NO\n");
        else printf("YES\n");
        return 0;
    }
    for(int i = 1; i < n; i++){
        int u, v;
        scanf("%d%d", &u, &v);
        addedge(u, v);
        addedge(v, u);
    }
    if(n == 2){
        if(A[1] == A[2]){
            printf("YES\n");
        }
        else{
            printf("NO\n");
        }
    }
    else{
        for(int i = 1; i <= n; i++){
            int num = 0;
            for(int j = head[i]; ~j; j = E[j].nxt){
                if(E[j].v != i){
                    num++;
                }
            }
            if(num > 1){
                //printf("*%d*\n", i);
                dfs(i, -1);
                break;
            }
        }
        printf("YES\n");
    }
    return 0;
}