AGC010 C cleaning

题意

这棵树有n个节点,节点i有ai个石头
每次操作选择两个叶子节点(u,v)(u不能等于v),移除u到v路径上的每一个节点的一块石头(包括u,v).
注意:如果这条路径上有一个节点没有石头,则不能进行操作.
能否通过上述操作将树上的石头移完
此处的叶子节点为度数为1的节点.

题解

对于非叶子节点，可以发现如果我们计算直接相连的边上经过的次数，这些次数加起来应该恰好是2*ai；而叶子结点，则是ai
然后我们选择一个非叶子节点为根 N=2时特判
然后我们从叶子往根部递推这条父边要求的容量
同时，如果说这个点直接相邻的边有一条边的次数大于了ai，那么就是非法的
就这样递推处理即可

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
using namespace std;
const int N=1e6+5;
typedef long long ll;
struct node{
    int u,v,nxt;
}edge[N*2];
int head[N],mcnt;
int du[N];
int a[N];
int n;
void add_edge(int u,int v){
    mcnt++;
    du[u]++;
    edge[mcnt].u=u;
    edge[mcnt].v=v;
    edge[mcnt].nxt=head[u];
    head[u]=mcnt;
}
void nosol(){
    puts("NO");
    exit(0);
}
void dfs(int u,int fa){
    if(du[u]==1)
        return ;
    int sum=0,mx=0;
    for(int i=head[u];i;i=edge[i].nxt){
        int v=edge[i].v;
        if(v==fa)
            continue;
        dfs(v,u);
        sum+=a[v];
        mx=max(mx,a[v]);
    }
    if(sum<a[u]||sum>2*a[u])
        nosol();
    if(mx>a[u])
        nosol();
    a[u]=2*a[u]-sum;
}
int main()
{
    //freopen("tree.in","r",stdin);
    //freopen("tree.out","w",stdout);
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    if(n==2){
        if(a[1]==a[2])
            puts("YES");
        else
            puts("NO");
        return 0;
    }
    for(int i=1;i<n;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        add_edge(u,v);
        add_edge(v,u);
    }
    int root=1;
    while(du[root]==1)
        root++;
    dfs(root,0);
    if(a[root])
        nosol();
    puts("YES");
}