没有上司的舞会(树形dp)

#没有上司的舞会
题目链接

题目描述

某大学有 n 个职员，编号为1…n。
他们之间有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。
现在有个周年庆宴会，宴会每邀请来一个职员都会增加一定的快乐指数 ri，但是呢，如果某个职员的直接上司来参加舞会了，那么这个职员就无论如何也不肯来参加舞会了。
所以，请你编程计算，邀请哪些职员可以使快乐指数最大，求最大的快乐指数。

输入格式

输入的第一行是一个整数 n。
第 2 到第(n+1) 行，每行一个整数，第 (i+1)行的整数表示 i 号职员的快乐指数 ri 。
第 (n + 2) 到第 2n行，每行输入一对整数l,k，代表 k 是 l 的直接上司。

输出格式

输出一行一个整数代表最大的快乐指数。

输入案例

7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5

输出案例

5

题目大意

输入人数.和每个人数的开心值,以及员工和他的上司.一个聚会如果他的直接上司来了,他就会不开心不会来了,上司的上司对他没有影响.如今让你计算怎么样才能使这次聚会的开心值最大.

算法分析

树形dp
dp[i][0]=dp[j][1];//dp[i][0]代表 i 没来,继承下属来了的情况
dp[i][1]=max(dp[j][0],dp[j][1]);//dp[i][1]代表 i 来了的情况下,判断是没来开心值大还是 i 来了 下属没来的情况开心值大;
使用链式前向星存图遍历所有的边.顺便记录入度为0的点,这个点为最大上司,由此点进行遍历

代码实现

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<map>
#include<set> 
#include<string>
#include<stack>
#include<queue>
#include<vector>
#include<math.h> 
#define inf 0x3f3f3f3f
using namespace std;
typedef long long int ll;
const int maxn=1e6+5;
typedef unsigned long long int ull;
ull p=131;
ll mod=1e9+7;
struct edge{
	int pre,to;
}e[maxn];
//e[i].pre 表示与第 i 条边同起点的上一条边 ;
//e[i].to 表示第 i 条边的终点
int head[maxn];//head 数组记录该 i 点的最后一条关联边
int f[maxn][2],a[maxn];//f[][0]是没来,f[][1]是来了
int rd[maxn]={0}; 
int cnt=0;
void add(int x,int y)
{
	e[++cnt].pre=head[x];
	e[cnt].to=y;
	head[x]=cnt;
} 
void dfs(int x)
{
	f[x][1]+=a[x];//自身带有欢乐值 
	for(int i=head[x];i;i=e[i].pre)
	{
		int s=e[i].to;//下属 
		dfs(s);//遍历下属 
		f[x][0]+=max(f[s][1],f[s][0]);//如果没来就只判断下属的来和没来的状态 
		f[x][1]+=f[s][0];//来了就继承加上下属没来的状态 
	}
}
int main()
{
	int n;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
			cin>>a[i];//纪录权值 
	}
	int a,b;
	for(int i=1;i<n;i++)
	{
		cin>>b>>a;//a代表上司,b代表下属 
		add(a,b);//由上司指向下属 
		rd[b]++;//有上司就入度加 1 
	}
	int rt;
	for(int i=1;i<=n;i++)	if(rd[i]==0) rt=i;//找入度为0的点 
	dfs(rt); 
	cout<<max(f[rt][0],f[rt][1])<<endl;
	return 0;
}