蓝桥杯 算法提高 结点选择（树形DP）

  算法训练 结点选择  

时间限制：1.0s   内存限制：256.0MB

       

问题描述

有一棵 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<bits/stdc++.h>
using namespace std;

int n;
int dp[100001][2];                  //dp[i][1] 表示选择节点i，dp[i][0] 表示不选择节点i
int List[100000+1][300];            //List[i][count]=num 节点i的第count个儿子是num

void Link(int x,int y){             //建树
    int i,j;
    i=j=0;
    while(List[x][i])   i++;        //如果节点x已经有儿子了，再增加一个儿子
    List[x][i]=y;

    while(List[y][j])   j++;        //同上，至于为什么这样连接，因为输入数据不会告诉你父子的关系
    List[y][j]=x;                   //因为不知道父子关系，那x有可能是y的儿子或者x是y的父亲
}
void tree_dp(int s,int f){          //树形DP，传入son father
    int t,k=0;                      //t存放s的儿子
	while(t=List[s][k++]){
		if(t!=f){                   //建树的时候会把儿子的父亲也当成了儿子的儿子~~~
			tree_dp(t,s);
            dp[s][1] += dp[t][0];
            dp[s][0] += max(dp[t][0],dp[t][1]);
        }
    }
}
int main(){
    while(scanf("%d",&n)!=EOF){
    	memset(dp,0,sizeof(dp));
    	memset(List,0,sizeof(List));
        for(int i=1;i<=n;i++)
            scanf("%d",&dp[i][1]);  //录入权值
        for(int i=1;i<n;i++){
            int x,y;
            scanf("%d%d",&x,&y);
            Link(x,y);
        }
        tree_dp(1,0);
        printf("%d\n",max(dp[1][1],dp[1][0]));
    }
    return 0;
}