CodeForces 77 C.Beavermuncher-0xFF（树形DP+贪心）

Description

给出一棵$n$个节点的树，每点有$val_i$个虫子，从树根$s$出发除虫，不能停留在一个节点，必须一直移动，每次只能向有虫子的节点移动，到达该节点后除去该节点的一个虫子，最后需要返回树根，问最多可以除掉多少虫子

Input

第一行一整数$n$表示节点数，之后输入$n$个整数$val_i$表示第$i$个节点的虫子数，之后输入$n-1$条树边，最后输入树根$s$

$(1\le n,val_i\le 10^5)$

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$节点最多可以剩余多少虫，以$eat(u),last(u)$分别表示这两个值。

为避免记重，在计算父亲节点时再考虑对于从父亲节点到当前节点再回去所除的两个虫。

$u$若是叶子节点，从父亲节点到叶子之后只会吃一个（在考虑父亲节点时计算），剩下$val_u-1$个都是剩的，故有$eat(u)=0,last(u)=val_u-1$

$u$若不是叶子节点，再不考虑其儿子节点时有$eat(u)=0,last(u)=val_u-1$，再考虑完$u$所有儿子节点$v_i$后，由于$u$点至多可以到$last(u)$个儿子节点去，故需要贪心的选择除虫多且剩余多的儿子节点，假设选取了$v$儿子，那么有$eat(u)+=eat(v)+2$，此处加的$2$即为从$u$到$v$再回到$v$所除的两个虫子，假设所选儿子节点中总剩余虫子有$sum$个，那么最多还可以从$u$到儿子再回来$min(last(u),sum)$次，每次都可以除掉两个虫子，故有$eat(u)+=2\cdot min(last(u),sum)$，然后将$min(last(u),sum)$从$last(u)$中减掉

注意，若$u$是根节点，那么$last(u)$初值为$val_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;
}