没有上司的舞会（树形dp）

原题链接

题目描述

输入描述

输出描述

输入样例

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

输出样例

5

数据说明

---

树形 dp 的板子题，如果从根向子结点进行遍历（因为每个结点只能有一个父亲，所以自上而下可以规避一些麻烦），显然对于每一个结点，选与不选只会影响它的子结点是否能够被选入，因此可用 f ( i , 0 / 1 ) 描述每一个状态，其中第一维代表该结点的编号，第二维表示是否被选入，其中 0 代表不选，1 代表可选。因为只会影响到子结点的状态，由此可得到状态转移方程：

$$f[i][0]=\Sigma max(f[i][0],f[i][1])$$
$$f[i][1]=\Sigma f[i][0]+a[i]$$

设根节点的根号为 n ，最后返回的值即为 f [ n ][ 1 ] 及 f [ n ][ 0 ] 之间的最大值，需要注意当仅有一个根节点时的特判。

参考代码

#include <bits/stdc++.h>
using namespace std;
const int N=6005;
struct edge{
	int v,next;
}e[N];
int head[N],n,cnt,f[N][2],ans,ru[N],vis[N];
void add(int u,int v){
	e[++cnt].v=v;//v记载的是该点本身的编号 
	e[cnt].next=head[u];
	head[u]=cnt;
}
void dfs(int k){
	vis[k]=1;
	for(int i=head[k];i;i=e[i].next){//枚举该节点的所有孩子 
		if(vis[e[i].v])
			continue;
		dfs(e[i].v);
		f[k][1]+=f[e[i].v][0];
		f[k][0]+=max(f[e[i].v][0],f[e[i].v][1]);//对于其每个孩子是否选择的最大值累加 
	}
	return; 
}


int main(){
	cin>>n;
	if(n==1){//特判仅有根节点情况
		int x;
		cin>>x;
		cout<<x;
		return 0;
	} 
	for(int i=1;i<=n;i++)
		cin>>f[i][1];//每个点的权值 直接转化为选择第i点的状态方程 
	for(int i=1;i<=n;i++){
		int l,k;
		cin>>l>>k;
		ru[l]=1;//即是有父亲的节点 
		add(k,l);
	}
	for(int i=1;i<=n;i++){
		if(ru[i]==0){//从根节点开始 
			dfs(i);
			cout<<max(f[i][1],f[i][0]);
			return 0;

		}
	}
	return 0;
}