HDU1520 Anniversary party (树状dp+DFS)

题意:

有个人准备开生日派对,邀请公司的所有员工,每个员工都有一定的权值,员工之间存在上下关系,邀请了上司就不能邀请下属,邀请了下属就不能邀请上司,求总权值最大。

题解:

构造一个树,来记录每个人之间的上下关系,之后根据数组dp[N][2],0表示不选择,1表示选择,先找到入读为0的根节点根据状态转移方程:dp[i][0] += max(dp[j][0], dp[j][1]);   dp[i][1] += dp[j][0];  最后输出最大值 (但是需要注意一些细节问题)

代码:

#include <iostream>
#include <cstring>
#include <queue>
#include <algorithm>
using namespace std;
const int N = 6000+5;
vector<int> tree[N];
int in[N], vis[N], dp[N][2], va[N];//0表示不选,1表示选 
void dfs(int x)
{
	vis[x] = 1;
	dp[x][0] = 0;
	dp[x][1] = va[x];
	int l=tree[x].size();
	for(int i=0;i<l;i++){
		int tmp=tree[x][i];
		if(vis[tmp]) continue;
		dfs(tmp);
		dp[x][0] += max(dp[tmp][0], dp[tmp][1]);
		dp[x][1] += dp[tmp][0];
	}
}
int main()
{
	int n, x, y;
	while(cin >> n){
		for(int i=0;i<=n;i++) tree[i].clear();//清除前一次的tree
		for(int i=1;i<=n;i++) cin >> va[i];//别写成i=0了 
		memset(in, 0, sizeof(in));
		memset(vis, 0, sizeof(vis));
		memset(dp, 0, sizeof(dp));
		while(cin >> x >> y, x+y) {
			tree[y].push_back(x);
			in[x] ++;
		}
		for(int i=1;i<=n;i++) {//不是i=0,根据人员编号确定 
			if(!in[i]){//寻找最大领导,即入度为0 
				dfs(i);
				cout << max(dp[i][0], dp[i][1]) << endl;
				break;
			}
		}	
	}	
	return 0;
}

阅读更多
想对作者说点什么? 我来说一句
相关热词

没有更多推荐了,返回首页