Description
给出一棵 n n 个节点的树,每点有个虫子,从树根 s s 出发除虫,不能停留在一个节点,必须一直移动,每次只能向有虫子的节点移动,到达该节点后除去该节点的一个虫子,最后需要返回树根,问最多可以除掉多少虫子
Input
第一行一整数表示节点数,之后输入 n n 个整数表示第 i i 个节点的虫子数,之后输入条树边,最后输入树根 s s
Output
输出最多除虫数量
Sample Input
5
1 3 1 3 2
2 5
3 4
4 5
1 5
4
Sample Output
6
Solution
考虑求出从父亲节点到 u u 节点后,在以节点为根节点的子树中最多可以除掉多少虫,除完最多虫的同时 u u 节点最多可以剩余多少虫,以分别表示这两个值。
为避免记重,在计算父亲节点时再考虑对于从父亲节点到当前节点再回去所除的两个虫。
u u 若是叶子节点,从父亲节点到叶子之后只会吃一个(在考虑父亲节点时计算),剩下个都是剩的,故有 eat(u)=0,last(u)=valu−1 e a t ( u ) = 0 , l a s t ( u ) = v a l u − 1
u u 若不是叶子节点,再不考虑其儿子节点时有,再考虑完 u u 所有儿子节点后,由于 u u 点至多可以到个儿子节点去,故需要贪心的选择除虫多且剩余多的儿子节点,假设选取了 v v 儿子,那么有,此处加的 2 2 即为从到 v v 再回到所除的两个虫子,假设所选儿子节点中总剩余虫子有 sum s u m 个,那么最多还可以从 u u 到儿子再回来次,每次都可以除掉两个虫子,故有 eat(u)+=2⋅min(last(u),sum) e a t ( u ) + = 2 ⋅ m i n ( l a s t ( u ) , s u m ) ,然后将 min(last(u),sum) m i n ( l a s t ( u ) , s u m ) 从 last(u) l a s t ( u ) 中减掉
注意,若 u u 是根节点,那么初值为 valu v a l u ,因为此时不用考虑从 s s 的父亲节点到这一步骤
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<ll,ll>P;
const int INF=0x3f3f3f3f,maxn=100005;
int n,val[maxn],s;
vector<int>g[maxn];
P dfs(int u,int fa)
{
if(g[u].size()==1&&u!=s)return P(0,val[u]-1);
vector<P>vec;
ll sum=0;
for(int i=0;i<g[u].size();i++)
{
int v=g[u][i];
if(v==fa)continue;
P temp=dfs(v,u);
sum+=temp.second;
vec.push_back(temp);
}
ll eat=0,last=val[u]-(u==s?0:1);
sort(vec.begin(),vec.end());
for(int i=vec.size()-1;last&&i>=0;last--,i--)eat+=vec[i].first+2;
eat+=2*min(last,sum);
last-=min(last,sum);
return P(eat,last);
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)scanf("%d",&val[i]);
for(int i=1;i<n;i++)
{
int u,v;
scanf("%d%d",&u,&v);
g[u].push_back(v),g[v].push_back(u);
}
scanf("%d",&s);
P ans=dfs(s,0);
printf("%I64d\n",ans.first);
return 0;
}