Algo-4-树形动态规划

问题描述
有一棵 n 个节点的树，树上每个节点都有一个正整数权值。如果一个点被选择了，那么在树上和它相邻的点都不能被选择。求选出的点的权值和最大是多少？
输入格式
第一行包含一个整数 n 。
接下来的一行包含 n 个正整数，第 i 个正整数代表点 i 的权值。
接下来一共 n-1 行，每行描述树上的一条边。
输出格式
输出一个整数，代表选出的点的权值和的最大值。
样例输入
5
1 2 3 4 5
1 2
1 3
2 4
2 5
样例输出
12
样例说明
选择3、4、5号点，权值和为 3+4+5 = 12 。
数据规模与约定
对于20%的数据， n <= 20。
对于50%的数据， n <= 1000。
对于100%的数据， n <= 100000。
权值均为不超过1000的正整数。

动态规划题。详细问题分析见代码注释。

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

using namespace std;

const int MAXN = 100000+10;

vector<int> e[MAXN];
int pow[MAXN];      //保存每个结点的权值 
int vis[MAXN];      //标记每个节点是否被遍历过 
int dp[MAXN][2];   //0表示不选这个点，1表示选择这个点

//动态规划是将两种选择的值都计算出来的 
//每一个结点都有选和不选两种情况，dp的列有两行，分别记录了选与不选时的最优解 
void dfs(int f){
    int next;
    //在进入递归之前，dp的值是当前此一节点的权值 
	//同时，这也是叶子结点最后返回的值(因为叶子结点不再含有未遍历过的子结点)
    dp[f][0] = 0; 
    dp[f][1] = pow[f];
    for(int i = 0;i<e[f].size();i++){ 
        next = e[f][i];
        if(vis[next] == 0){
            vis[next] = 1;//递归这个点，走过置1 
            dfs(next);
            //在递归完子结点之后，做如下操作
            dp[f][1] += dp[next][0];  //如果f点选择，那么和f相连的点就只能先不选 不选择子结点的情况，应该加给选择了本节点的情况 
			dp[f][0] += max(dp[next][1],dp[next][0]);  //如果f点不选择，那就相邻结点就可以选，选的话就选相邻结点的最优解 
			//所以在函数返回的时候，dp中保存的是到目前为止最优的权值之和 
        }
    }
}

int main(){
    int n,a,b;
    cin >> n;
    memset(vis,0,sizeof(vis));
    for(int i = 1;i<=n;i++) {
        cin >> a;
        pow[i] = a;
    }
    //邻接表存储
    for(int i = 0;i < n-1;i++){
        cin >> a >> b;
        e[a].push_back(b);
        e[b].push_back(a); 
    }
    vis[1] = 1;
    //从第一个点开始递归 
    dfs(1);
    cout << max(dp[1][1],dp[1][0]);
    return 0;
}