树形dp入门题，上司的舞会

最新推荐文章于 2022-10-14 23:35:47 发布

开学了8

最新推荐文章于 2022-10-14 23:35:47 发布

阅读量116

点赞数

本文链接：https://blog.csdn.net/luker6/article/details/119494101

版权

该博客介绍了一个关于图论的问题——没有上司的舞会。作者通过状态转移方程和深度优先搜索算法（DFS）来解决这个问题。在给定的员工关系网络中，每个节点代表一个员工，边表示上下级关系。目标是找到最大数量的员工，他们之间没有直接或间接的上下级关系。博客提供了完整的C++代码实现，并在洛谷平台上运行，具有时间和内存限制。

摘要由CSDN通过智能技术生成

遍历根节点，当前节点为u，一次遍历儿子节点x
状态转移方程：
$\\ dp[u][1] += dp[x][0]$

// Problem: P1352 没有上司的舞会
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P1352
// Memory Limit: 128 MB
// Time Limit: 1000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include<iostream>
#include<cstdio>
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<climits>
#include<queue>
#include<map>
#include<set>
#include<sstream>
#include<cassert>
#include<bitset>
#include<list>
#include<unordered_map>
using namespace std;
#define lowbit(x) (x&-x)
#define pf(a) printf("%d\n",a)
#define mem(x,y) memset(x,y,sizeof(x))
#define dbg(x) cout << #x << " = " << x << endl
#define rep(i,l,r) for(int i = l; i <= r; i++)
#define fep(i,a,b) for(int i=b;i>=a;--i)
typedef long long ll;
typedef unsigned long long ull;

// const int inf=0x3f3f3f3f;
const int N=6100;
vector<int> v[N];
int a[N],p[N],dp[N][2];
int n;

void dfs(int u)
{
	dp[u][1] = a[u];
	for(auto x : v[u])
	{
		dfs(x);
		dp[u][1] += dp[x][0];
		dp[u][0] += max(dp[x][0], dp[x][1]);
	}
}

void solve()
{
	cin >> n;
	rep(i,1,n) cin >> a[i];
	rep(i,1,n-1)
	{
		int a, b;
		cin >> a >> b;
		p[a] = b;
		v[b].push_back(a);
	}
	int root = 0;
	rep(i,1,n)
		if(!p[i]){
			root = i;
			break;
		}
	dfs(root);
	cout << max(dp[root][0], dp[root][1]) << endl;
}

int main()
{

	ios::sync_with_stdio(false);cin.tie(0);
	// int Case;cin >> Case;
	// while(Case--)
		solve();
	return 0;
}