CF77C Beavermuncher-0xFF

一、题目

点此看题

二、解法

对于每一个点，要考虑他到子树的走法。由于这个点要被根走到才能产生贡献，所以先预留一个海狸让父亲能走过来（根就特判），然后贪心地走到贡献最大的子树中去（因为可能海狸个数不够访问所有子树）

如果还有剩余，那么就跟儿子来回走刷贡献。

#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
const int M = 100005;
#define int long long
int read()
{
    int x=0,flag=1;char c;
    while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
    while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
    return x*flag;
}
int n,rt,tot,f[M],a[M],dp[M];
struct edge
{
	int v,next;
}e[2*M];
int cmp(int a,int b)
{
	return a>b;
}
void dfs(int u,int fa)
{
	vector<int> vc;
	for(int i=f[u];i;i=e[i].next)
	{
		int v=e[i].v;
		if(v==fa) continue;
		dfs(v,u);
		vc.push_back(dp[v]);
	}
	a[u]-=(u!=rt);
	if(!vc.size()) return ;
	sort(vc.begin(),vc.end(),cmp);
	for(int i=0;i<vc.size();i++)
	{
		if(a[u]<=0) break;
		a[u]--;
		dp[u]+=vc[i]+2;
	}
	for(int i=f[u];i;i=e[i].next)
	{
		int v=e[i].v;
		if(v==fa) continue;
		dp[u]+=2*min(a[u],a[v]);
		a[u]-=min(a[u],a[v]);
	}
}
signed main()
{
	n=read();
	for(int i=1;i<=n;i++)
		a[i]=read();
	for(int i=1;i<n;i++)
	{
		int u=read(),v=read();
		e[++tot]=edge{v,f[u]},f[u]=tot;
		e[++tot]=edge{u,f[v]},f[v]=tot;
	}
	rt=read();
	dfs(rt,0);
	printf("%lld\n",dp[rt]);
}