题目描述:每个节点有权值,不能同时选择相邻的两个节点,问最后选的最大价值。
贪心的想法是要不选子节点,要不选父亲节点,可是有可能最优的是两个节点都不选。
那么就是类似01背包选择路线,来个dp[i][2],0表示不选,1表示选,那么从叶子节点往根节点不断更新就好了,状态转移方程
dp[i][1]=sum(dp[j][0])
dp[i][0]=sum(max(dp[j][1].dp[j][0])
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<vector>
using namespace std;
vector<int>Edge[6010];
int visit[6010],dp[6010][2],val[6010];
void dfs(int u)
{
if(visit[u]) return;
visit[u]=1;
for(int i=0;i<Edge[u].size();i++)
{
int v=Edge[u][i];
if(!visit[v])
{
dfs(v);
dp[u][0]+=max(dp[v][1],dp[v][0]);
dp[u][1]+=dp[v][0];
}
}
dp[u][1]+=val[u];
}
void init()
{
memset(dp,0,sizeof(dp));
}
int main()
{
int n;
while(~scanf("%d",&n))
{
for(int i=1;i<=n;i++)
scanf("%d",&val[i]);
for(int i=1;i<=n;i++)
{
Edge[i].clear();
visit[i]=0;
}
int a,b;
scanf("%d %d",&a,&b);
while(a+b!=0)
{
Edge[a].push_back(b);
Edge[b].push_back(a);
scanf("%d %d",&a,&b);
}
init();
dfs(1);
printf("%d\n",max(dp[1][0],dp[1][1]));
}
}