Anniversary party POJ - 2342 (树形DP)

时间:2023-12-04 22:22:50

题目链接:

 POJ - 2342

题目大意:给你n个人,然后每个人的重要性,以及两个人之间的附属关系,当上属选择的时候,他的下属不能选择,只要是两个人不互相冲突即可。然后问你以最高领导为起始点的关系网的重要性最大。

具体思路:简单树形DP,

dp[i][0]表示当前i点不选择,那么dp[i][0] =  sum( max(dp[to][1] ,dp[to][0] ) )(to为i的子节点)。

dp[i][1]表示当前i点选择, 那么dp[i][1] = dp[i][1] + sum(dp[to][0])(to为i的子节点)。

反思:可能会有多个起点的情况

AC代码:

 #include<iostream>
#include<stdio.h>
#include<cmath>
#include<vector>
#include<cstring>
#include<string>
using namespace std;
# define ll long long
# define inf 0x3f3f3f3f
const int maxn = 2e5+;
int dp[maxn][];
int in[maxn];
vector <int> Edge[maxn];
int vis[maxn];
void dfs(int u)
{
vis[u]=;
for(int i=; i<Edge[u].size(); i++)
{
int to=Edge[u][i];
if(vis[to])
continue;
dfs(to);
dp[u][]=dp[u][]+dp[to][];
dp[u][]=dp[u][]+max(dp[to][],dp[to][]);
}
}
int main ( )
{
int n;
scanf("%d", &n);
for(int i= ; i <=n ; i++ )
{
scanf("%d", &dp[i][] );
}
int st,ed;
while(~scanf("%d %d",&st,&ed)&&(st+ed))
{
Edge[ed].push_back(st);
in[st]++;
}
int root;
for(int i=; i<=n; i++)
{
if(!in[i])
{
root=i;
dfs(root);
break;
}
}
printf("%d\n",max(dp[root][],dp[root][]));
return ;
}