luogu1122:最大子树和

题目连接

  • 树形DP专题

题目大意

  • 知道一棵树,每个点有权重;
  • 删掉任意个点,保留以1为根的子树;
  • 使得权值和最大。

题目分析

  • 树形DP的入门题,考核树上的递归更新。

解题思路

  • 一个dfs完成

参考代码
//luogu1122-最大子树和
#include
using namespace std;
const int N=20000;
int a[N],las[N],f[N];
int n,ans,len;
struct nod{
	int nex,to;
}e[N];
void add(int x,int y){
	len++;
	e[len].nex=las[x];
	e[len].to=y;
	las[x]=len;
}
void dfs(int x,int fa){	//递归找答案 
	f[x]=a[x];
	for(int i=las[x];i;i=e[i].nex){
		int y=e[i].to;
		if(y!=fa){
			dfs(y,x);
			f[x]+=max(0,f[y]);	//回溯更新 
		}
	}
	ans=max(ans,f[x]);
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	int x,y;
	for(int i=1;i<n;i++){
		scanf("%d %d",&x,&y);
		add(x,y);	//构图 
		add(y,x);
	}
	dfs(1,0);	//当前节点是 1,父亲为 0 
	cout<<ans;
	return 0;
} 

你可能感兴趣的:(树形DP,DP)