题意:给定一棵树,每个节点的权值为Ai,每次可以挑选两个叶子节点,将两个叶子节点之间路径上的权值-1。问最后能不能使得整棵树的权值为0。
解法:验证解的存在性,就去试着找出一组可行解。这题从最末端的情况开始想就可以很容易想出dp。
随便找个非叶子节点,将其作为整棵树的根,现在对这个有根树进行分析。
首先可以将路径进行归类,对于一个非叶子节点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的时候,要能真正画出这样的情况。观察发现,仅当儿子向上传递的权值分配很畸形的时候才画不出来。设tm为儿子中向上传递权值的最大值,当tm>Ai的时候,就不能画出x1条第一种路径,x2条第二种路径。动手模拟一下就可以看出来。
最后,看一下整棵树的根节点是不是x1等于0,若是的话,证明找到了可行解,因为经过根节点的只能是第二种边。
#include <cstdio>
#include <vector>
#include <cstdlib>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned int uii;
const int maxn=100005;
int n,a,b;
ll A[maxn];
vector<int> tr[maxn];
ll dfs(int u,int fa) {
if (tr[u].size()==1)
return A[u];
ll tot=0,x2=0,x1=0,mm,tm=0;
for (uii i=0;i<tr[u].size();++i) {
if (tr[u][i]==fa)
continue;
mm=dfs(tr[u][i],u);
tm=max(tm,mm);
tot+=mm;
}
x2=tot-A[u];
x1=A[u]-x2;
if (x1<0||x2<0||A[u]<tm||(x1>0&&fa==-1)) {
puts("NO");
exit(0);
}
return x1;
}
int main()
{
scanf("%d",&n);
for (int i=1;i<=n;++i)
scanf("%lld",&A[i]);
if (n==1) {
puts(A[1]?"NO":"YES");
return 0;
}
for (int i=0;i<n-1;++i) {
scanf("%d%d",&a,&b);
tr[a].push_back(b);
tr[b].push_back(a);
}
if (n==2)
puts(A[1]==A[2]?"YES":"NO");
else {
for (int i=1;i<=n;++i)
if (tr[i].size()>1) {
dfs(i,-1);
break;
}
puts("YES");
}
return 0;
}