P1352【模板】【链式前向星】【树形dp】【没有上司的舞会】

最新推荐文章于 2023-02-26 08:17:26 发布

while WA er

最新推荐文章于 2023-02-26 08:17:26 发布

阅读量147

点赞数

分类专栏：树形dp 文章标签：深度优先搜索图论最大路径 C++ 算法

本文链接：https://blog.csdn.net/peacesir/article/details/119209253

版权

树形dp 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

没有很多的讲解= = 注意用法。

题目入口

#include <bits/stdc++.h>
#define re register int
using namespace std;
const int N=1e6;
int n;
struct node
{
	int ver,next;
}e[N];
int a[N],in[N];
int head[N],tot;
int f[N][2];
void add(int x,int y)
{
	e[++tot].ver=y;
	e[tot].next=head[x];
	head[x]=tot;
}
void dfs(int x,int pre)
{
	f[x][0]=0,f[x][1]=a[x];
	for(re i=head[x];i;i=e[i].next)
	{
		int y=e[i].ver;
		if(y==pre)  continue;
		dfs(y,x);
		f[x][0]+=max(f[y][0],f[y][1]);
		f[x][1]+=f[y][0];
	}
}
void solve()
{
	cin>>n;
	for(re i=1;i<=n;i++)  scanf("%d",&a[i]);
	for(re i=1;i<n;i++) 
	{
		int x,y;
		scanf("%d%d",&x,&y);
		add(x,y);add(y,x);
		in[y]++;
	}
	int root;
	for(re i=1;i<=n;i++)  if(!in[i])  root=i;
	dfs(root,root);
	cout<<max(f[root][0],f[root][1])<<endl;
}
int main() {
	solve();
	return 0;
}