初学动态规划,然后自己试着做了这一道题 ,发现思路完全错误……于是就花了好几个小时百度,终于对这道题有了一点理解!!!下面是思路:
定义一个二维数组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的父亲的情况)画个图其实更好理解。