【ybt】【树形DP例1】树上求和

树上求和

Link

---

解题思路

模板题。
设 $f_{i,0}$ 表示第 $i$ 个点不选的最大值， $f_{i,1}$ 表示第 $i$ 个点选的最大值。
$dfs$ 跑图，回溯求和。

code

#include<iostream>
#include<cstdio>
using namespace std;

int n;
int a[6010];
int f[6010][2];
int hd[6010],tot;

struct abc{
	int to,nxt;
}b[12010];

void add(int x,int y)
{
	b[++tot]=(abc){y,hd[x]};
	hd[x]=tot;
}

void dfs(int x,int fa)
{
	f[x][1]=max(a[x],0);
	for(int i=hd[x];i;i=b[i].nxt)
	{
		int y=b[i].to;
		if(y!=fa)
		{
			dfs(y,x);
			f[x][1]+=max(f[y][0],0);
			f[x][0]+=max(f[y][1],0);
		}
	}
}

int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for(int i=1;i<n;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		add(x,y),add(y,x);
	}
	dfs(1,0);
	printf("%d",max(f[1][0],f[1][1]));
}