AtCoder Grand Contest 010 C - Cleaning

https://atcoder.jp/contests/agc010/tasks/agc010_c

这题的关键是每个点向上的路径和儿子们自己匹配的路径可以解方程解出来啊

没想到啊= =

f[u]表示u向父节点的路径，pr为子树内部匹配的路径

sum=所有子树向上的路径，其中有pr对自动配对了，f[u]+2*pr=sum,pr+f[u]=a[u]

能凑出pr对来的充要条件是最大的那个f[v]*2<=sum

然后要满足f[rt]=0，pr>=0,f[u]>=0,a[u]>=0，就行了

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int maxl=3e5+10;

int n,m,k,cnt,tot,cas,ans;
int a[maxl],deg[maxl];ll dp[maxl],f[maxl];
bool vis[maxl];
vector<int> e[maxl];
char s[maxl];

inline void prework()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for(int i=1;i<n;i++)
	{
		int u,v;scanf("%d%d",&u,&v);
		e[u].push_back(v);deg[u]++;
		e[v].push_back(u);deg[v]++;
	}
}

inline void dfs(int u,int fa)
{
	if(deg[u]==1)
	{	
		f[u]=a[u];
		return;
	}
	ll sum=0,mx=0;
	for(int v:e[u])
	if(v!=fa)
	{	
		dfs(v,u);
		sum+=f[v];mx=max(mx,f[v]);
	}	
	ll pr=sum-a[u];
	f[u]=sum-2*pr;
	if(f[u]>a[u] || min(sum/2,sum-mx)<pr)
		ans=0;
}

inline void mainwork()
{
	if(n==2)
	{
		ans=(a[1]==a[2]);
		return;
	}
	int rt;
	for(int i=1;i<=n;i++)
	if(deg[i]>1)
	{
		rt=i;
		break;
	}
	ans=1;
	dfs(rt,0);
	if(f[rt]!=0) ans=0;
}

inline void print()
{
	puts(ans?"YES":"NO");	
}

int main()
{
	int t=1;
	//scanf("%d",&t);
	for(cas=1;cas<=t;cas++)
	{
		prework();
		mainwork();
		print();
	}
	return 0;
}