题意:
给你一颗树,树上每个节点有a[i]个糖果(其实是翻译是海狸,但是吃海狸有点恶心。。。。)可以吃,现给出起点,每次走到一个有糖果的节点上可以吃一个糖果,没有糖果的节点不走,问你能获得最大的糖果数是多少。
思路:
对于每个点,如果我们能够知道它所有儿子节点所能获得的最大糖果数和还剩的糖果数,那么我们就可以做如下两种操作:
1.在所有儿子节点中最多选择a[i]-1个来获得其糖果数。
2.如果上面操作完成后,该节点还有糖果剩余则遍历其所有儿子节点的剩余糖果数,然后让它来回走来获得其糖果数量。
然后对其儿子也进行如上操作,不断递归下去,从局部到整体,获得最后的总糖果数量。树形DP思想。
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
typedef __int64 LL;
const int MAX=1e5+5;
int n,s;
LL a[MAX],dp[MAX];
vector<int> gr[MAX];
bool cmp(LL a,LL b){
return a>b;
}
void DFS(int u,int f){
int nc=gr[u].size();
if(nc==1&&u!=s){//判断是否为叶子
dp[u]=1;
a[u]--;
return;
}
vector<LL> son;//储存所有的儿子的值,等下从中选择最大的来吃
for(int i=0;i<nc;i++){
int v=gr[u][i];
if(v!=f){
DFS(v,u);
son.push_back(dp[v]);
}
}
if(u!=s){
dp[u]=1;
a[u]--;
}
else{
dp[u]=0;
}
sort(son.begin(),son.end(),cmp);
for(int i=0;i<son.size();i++){
if(a[u]){
dp[u]+=son[i]+1;//每次向下走要加1个
a[u]--;
}
else break;
}
for(int i=0;i<nc;i++){
int v=gr[u][i];
if(v!=f){
if(a[u]){
dp[u]+=min(a[u],a[v])*2;
a[u]-=min(a[u],a[v]);
}
else break;
}
}
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%I64d",&a[i]);
}
int u,v;
for(int i=1;i<n;i++){
scanf("%d%d",&u,&v);
gr[u].push_back(v);
gr[v].push_back(u);
}
scanf("%d",&s);
DFS(s,-1);
printf("%I64d\n",dp[s]);
return 0;
}