题目
在大家的三连助攻下,TT 一举获得了超级多的猫咪,因此决定开一间猫咖,将快乐与大家一同分享。并且在开业的那一天,为了纪念这个日子,TT 在猫咖门口种了一棵苹果树。
一年后,苹果熟了,到了该摘苹果的日子了。
已知树上共有 N 个节点,每个节点对应一个快乐值为 w[i] 的苹果,为了可持续发展,TT 要求摘了某个苹果后,不能摘它父节点处的苹果。
TT 想要令快乐值总和尽可能地大,你们能帮帮他吗?
结点按 1~N 编号。
第一行为 N (1 ≤ N ≤ 6000) ,代表结点个数。
接下来 N 行分别代表每个结点上苹果的快乐值 w[i](-128 ≤ w[i] ≤ 127)。
接下来 N-1 行,每行两个数 L K,代表 K 是 L 的一个父节点。
输入有多组,以 0 0 结束。
每组数据输出一个整数,代表所选苹果快乐值总和的最大值。
7
1
1
1
1
1
1
1
1 3
7 4
2 3
4 5
6 4
3 5
0 0
5
思路
这题和上司舞会问题相似,使用树型DP解决。
注意:记得初始化
代码
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn=7000;
int w[maxn];
vector<int> v[maxn];//存储每个节点的孩子节点
int dp[maxn][2];
int root[maxn];
int mark[maxn];
void init()
{
for(int i=0;i<maxn;i++)
{
w[i]=0;v[i].clear();
root[i]=1;
mark[i]=0;
}
}
void solve(int x)
{
mark[x]=1;
if(v[x].size()==0)
{
dp[x][1]=w[x];
dp[x][0]=0;
}
int tmp=0,tmp2=0;
for(int i=0;i<v[x].size();i++)
{
if(mark[v[x][i]]==0)
{
solve(v[x][i]);
//上司参加 ,员工不参加
tmp+=dp[v[x][i]][0];
//上司不参加,下属可参加可不参加
tmp2+=max(dp[v[x][i]][0],dp[v[x][i]][1]);
}
}
dp[x][1]=w[x]+tmp;
dp[x][0]=tmp2;
}
int main()
{
int n;
while(cin>>n&&n!=0)
{
init();
for(int i=1;i<=n;i++)
cin>>w[i];
for(int i=1;i<n;i++)
{
int l,k;
cin>>l>>k;
root[l]=0;
v[k].push_back(l);
}
int r=0;
//得到根节点
for(int i=1;i<=n;i++)
if(root[i]==1)
{
r=i;
break;
}
solve(r);
cout<<max(dp[r][1],dp[r][0])<<endl;
}
return 0;
}