Codeforces 77C 树形DP

/*
给出一颗树，每颗树上有ki个虫子
一个熊从根节点出发，沿着边走，每去一个地方吃一个虫子（去的地方少于1个虫子就不能去），最后必须要回到根，问最多能吃多少虫子

解法：
这题自己想出来的，，，但是细节没有想的很清楚，看了过了的代码才明白哪里错了
首先画几个样例大概能明白这个问题可以递归解决，对每个节点v可以分两步
1.从v出发去v的每个孩子，孩子处理完后，拿到孩子的收益
2.所有孩子的收益拿完后，可以在v和v的孩子间循环往复直到消耗完v这个点剩下的虫子

明白这两步剩下好好想想细节就可以了
dfs返回的东西为pair<int,int>，第一个int表示该节点的子树的收益，第二个int表示该点的剩余虫子
该节点子树的收益定为该节点从儿子们拿到的收益和该节点和儿子们剩余虫子们对拼的收益
结点v的剩余虫子定义为：1.去除从节点v的父亲到节点v的1个虫子 2.去掉节点v和节点v的孩子的剩余虫子对拼的消耗
剩下的便是节点v的剩余虫子，对叶子节点来说，剩余虫子便是初始虫子-1


*/
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <iostream>
#include <set>
using namespace std;
#define ll long long
#define pb push_back
#define PII pair<ll,ll>
#define ALL(x) (x).begin(),(x).end()
const int MAXN = 101111;
int k[MAXN];
vector<int> G[MAXN];
int root;
ll ans;

PII dfs(int now,int fa){
    if(G[now].size() == 1 && now != root) return PII(0,k[now]-1);
    vector<PII > vc;//不能申请一个全局的再每次clear
    for(int i=0;i<G[now].size();i++){
        int to = G[now][i];
        if(to == fa) continue;
        vc.pb(dfs(to, now));
    }
    ll left=0;for(int i=0;i<vc.size();i++) left += vc[i].second;

    sort(ALL(vc));
    reverse(ALL(vc));

    ll ene = 1LL*(k[now] - (now != root));

    ll value=0;
    for(int i=0;i<vc.size() && ene;ene--,i++) value += vc[i].first + 2;
    value += min(ene, left)*2;
    ene -= min(ene, left);
    //cout<<now<<" "<< value<<" "<<ene<<endl;
    return PII(value, ene);
}
int main()
{
    //freopen("in.txt","r",stdin);
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&k[i]);
    for(int i=0;i<n-1;i++){
        int u,v;scanf("%d%d",&u,&v);
        G[u].pb(v);
        G[v].pb(u);
    }
    scanf("%d",&root);
    PII res = dfs(root,-1);
    printf("%I64d\n",res.first);
    return 0;
}