Description
话说一个公司的一些然要去参加一个party,每个人有一个愉悦值,而如果某个人的直接上司在场的话会非常扫兴,所以避免这样的安排,问给出n个人,每个人的愉悦值以及他们的上司所属关系,问你让那些人去可以让总的愉悦值最大,并求出这个值
Input
第一行为总人数n,之后n行为每个人的愉悦值,然后每行两个整数a和b表示所属关系b是a的上司,以0 0结束所属关系输入
Output
输出最大总愉悦值
Sample Input
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0
Sample Output
5
Solution
这个公司的人事关系可以根据给出的数据得到一个树,最上面的是最高层,往下依次,我们要做的就是在树的节点处进行dp。
用dp[i][0]表示当前i这个人不去,dp[i][1]表示当前i这个人去
那么dp[fa[i]][1]+=dp[i][0]
而 dp[fa[i]][0]+=max(dp[i][0],dp[i][1])
Code
#include<cstdio>
#include<iostream>
#include<vector>
using namespace std;
#define maxn 6666
int a[maxn],fa[maxn],dp[maxn][2];
vector<int> vec[maxn];
int n;
void dfs(int root)
{
dp[root][1]=a[root];
int len=vec[root].size();
for(int i=0;i<len;i++)
dfs(vec[root][i]);
for(int i=0;i<len;i++)
{
dp[root][0]+=max(dp[vec[root][i]][1],dp[vec[root][i]][0]);//这个人不去
dp[root][1]+=dp[vec[root][i]][0];//这个人去
}
}
int main()
{
while(~scanf("%d",&n))
{
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
//初始化
vec[i].clear();
fa[i]=-1;
dp[i][0]=dp[i][1]=0;
}
int u,v;
while(scanf("%d%d",&u,&v),u||v)
{
vec[v].push_back(u);
fa[u]=v;
}
int temp=1;
while(fa[temp]!=-1)//找到根节点
temp=fa[temp];
dfs(temp);//从根节点开始搜索
int ans=max(dp[temp][0],dp[temp][1]);
printf("%d\n",ans);
}
return 0;
}