题意:公司内的员工关系表是一棵树,每个员工有一个权值,在举行年会时一个员工与他的直属上司不能同时出现(即直接相邻的节点不能同时选取),求最大权值。
思路:树形dp入门题。
选取节点 1 为根,对于每个节点 u 和它的子节点 v:
如果选取节点 u,则节点 v 必不能被选:
dp[u][1] += dp[v][0]
如果不选取节点 u,则选择节点 v 贡献大的一个(题目的要求是不能同时选取,当然可以同时不选取):
dp[u][0] += max(dp[v][0], dp[v][1])
tips:题目还有一个坑点是多组数据没有指明。
AC代码:
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=6e3+7;
struct node{
int to,nxt;
}e[maxn*2];
int a[maxn],tot,head[maxn],dp[maxn][2];
void add(int u,int v){
e[tot].nxt=head[u];
e[tot].to=v;
head[u]=tot++;
return;
}
void add_edge(int u,int v){
add(u,v);
add(v,u);
return;
}
void dfs(int rt,int p){
for(int i=head[rt];~i;i=e[i].nxt){
int to=e[i].to;
if(to==p) continue;
dfs(to,rt);
dp[rt][1]+=dp[to][0];
dp[rt][0]+=max(dp[to][0],dp[to][1]);
}
dp[rt][1]+=a[rt];
return;
}
int main(void){
int n;
while(~scanf("%d",&n)){
tot=0;
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
dp[i][0]=dp[i][1]=0;
head[i]=-1;
}
int u,v;while(~scanf("%d%d",&u,&v)){
if(!u&&!v) break;
add_edge(u,v);
}
dfs(1,0);
printf("%d\n",max(dp[1][0],dp[1][1]));
}
return 0;
}