算法训练 结点选择

算法训练 结点选择
时间限制：1.0s 内存限制：256.0MB
提交此题 锦囊1 锦囊2
问题描述
有一棵 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的正整数。

这道题在建树的时候不知道咋弄，参考好多，终于看明白采用了孩子链表表示法存储树的结构比较好，如果采用二维数组，则会浪费较多空间。一起加油

动态规划, 用dp[i][0]表示不选择i点时，i点及其子树能选出的最大权值，dp[i][1]表示选择i点时，i点及其子树的最大权值。

状态转移方程:

对于叶子节点
dp[i][0] = 0, dp[i][1] = i点权值

对于非叶子节点i
dp[i][0] += max(dp[j][0], dp[j][1]) (j是i的儿子)
dp[i][1] += i点权值 + dp[j][0] (j是i的儿子)
最大权值即为max(dp[1][0], dp[1][1])

#include<stdio.h> 
#include<string.h>
#include<stdlib.h>
#define Max_ 100005
#define max(a,b) a>b?a:b
int dp[Max_][2],head[Max_],m=0;//head[]记录结点所指的下一条边 ，dp[]记录当前的最优值 

struct Edge{
    int tonode;//记录这条边指向的结点 
    int nextedge;//记录这条边指向的结点所指向的下一条边 
}edge[Max_*2];

addedge(int fromnode,int tonode){//将指向tonode结点的那条边加入边集 棵树 
    edge[m].tonode=tonode;//这条边指向的结点 
    edge[m].nextedge=head[fromnode];//fromnode结点所指向的下一条边,即，边edge[m]与边edge[m].nextedge在此棵树的同一层, 
    head[fromnode]=m++;// fromnode指向新生成的这条边 
}

void dfs(int fromnode,int prenode){//类似dfs 
     int i;
     for(i=head[fromnode];i!=-1;i=edge[i].nextedge){//head[fromnode]这条边的同一层进行遍历 
        if(edge[i].tonode==prenode){//此结点已经遍历过 
              continue;
        }
        dfs(edge[i].tonode,fromnode);//继续遍历edge[i].tonode结点相连的下一条边，直到出现叶子结点 
        dp[fromnode][1]+=dp[edge[i].tonode][0];//此层取，下一层不取 
        dp[fromnode][0]+=max(dp[edge[i].tonode][0],dp[edge[i].tonode][1]);//此层部取，下一层或取不取    
     }


}

int main(){
    memset(dp,0,sizeof(dp));
    int n,i,fromnode,tonode,result;
    scanf("%d",&n);
    for(i=1;i<=n;i++){//记录最初各个结点的权值 
        scanf("%d",&dp[i][1]);
    }
    for(i=0;i<=n;i++){
        head[i]=-1;// 最开始各节点并没有指向任何一边，标记head[i]为-1 
    }
    for(i=1;i<n;i++){
        scanf("%d%d",&fromnode,&tonode);
        //用两条边进行表示，则父结点和孩子结点都会在同一层上，两层中任意相连的两节点最多只能取一个 
        addedge(fromnode,tonode);//将指向tonode结点的那条边加入边集 
        addedge(tonode,fromnode);//将指向fromnode结点的那条边加入边集
    }
    dfs(1,0);
    result=max(dp[1][0],dp[1][1]);
    printf("%d",result);
    return 0;
}