蓝桥杯 ALGO-4 结点选择（树形动态规划）

初学动态规划，然后自己试着做了这一道题 ，发现思路完全错误……于是就花了好几个小时百度，终于对这道题有了一点理解！！！下面是思路：

定义一个二维数组dp，dp[i][0]表示结点i不被选择时 ， i点及其子树能选出的最大权值；dp[i][1]表示结点选择i时，i点及其子树能选出的最大权值。

对于叶子结点dp[i][0] = 0 ； dp[i][1] = i（也可以说是此时数组所存的权值）;

对于非叶子结点dp[i][0] =∑ MAX(dp[j][1],dp[j][0]);

dp[i][1] = i + ∑dp[j][0];  知道了大体思路就可以开始递推了，下面是代码（参考网上）：

#include <iostream>
#include <vector>
#include <cstdio>

using namespace std; 

vector<int> vec[100001];
int value[100001][2];

void dp(int a,int b){
	for( int i = 0 ; i < vec[a].size() ; i++ ){
		int k = vec[a][i];
		if(k!=b){
			dp(k,a);
			value[a][1] += value[k][0]; 
			value[a][0] += (value[k][0]>value[k][1]?value[k][0]:value[k][1]);
		}
	}
}

int main(){
	int n;
	cin>>n;
	
	for( int i = 1 ; i <= n ; i++ ){
		int val;
		cin>>val;
		value[i][1] = val;
	}
	
	for( int i = 1 ; i < n ; i++ ){
		int x,y;
		cin>>x>>y;
		vec[x].push_back(y);
		vec[y].push_back(x);
	}
	
	for( int i = 1 ; i <= n ; i++ ){
		if(vec[i].size()==1)/*从叶子节点开始遍历*/{
			dp(i,-1);
			cout<<(value[i][0]>value[i][1]?value[i][0]:value[i][1]);
			break;
		}
	}
	
	return 0;
}

整个写完后对第13行的if（k！= b）有点不理解，最后想了想感觉其实这句代码其实就是为了解决回溯问题（防止出现a的孩子变成a的父亲的情况）画个图其实更好理解。