蓝桥杯 算法训练 结点选择 DP（SDNU 1482.结点选择）

结点选择
问题描述
有一棵 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的方式对所有节点遍历，从最下面的子结点开始找到当前节点的最佳方案，然后一层一层向上遍历，最后找到根节点得出最佳的方案。

    这里要确定所有点的各种方案全部遍历，所以直接用DFS的方法来进行遍历。

    我们假设dp[i][j]来表示i点及其所有子节点的最大总权值，j取值范围为0或1,0代表这个点的权值不使用，1代表这个点使用权值。

①对于叶子结点

    如果不使用的话则权值为0，dp[i][0]=0，如果使用的话就是这个点的权值dp[i][1]=v[i]。

②对于非叶子结点

    如果不使用的话，权值就是前一个点总权值最大的一个，即dp[i][0]=dp[i][0]+max(dp[k][0],dp[k][1])。如果使用，则前一个点肯定不可以使用，则dp[i][1]=dp[i][1]+dp[k][0]。（k表示前一个点）

即

①对于叶子结点
dp[i][0]=0
dp[i][1]=v[i]
②对于非叶子结点
dp[i][0]=dp[i][0]+max(dp[k][0],dp[k][1])

dp[i][1]=dp[i][1]+dp[k][0]

（其实仔细观察会发现最后①和②得出的结果是相同的，所以只写②就能同时解决两种情况）

    下面AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

struct Edge
{
    int to,next;//分别表示到达的点，和以这个点为出发点的另一条边
};

Edge e[200005];//一条边记录两次，分别以一个点做记录
int head[100005];
int m;
int dp[100005][5];

void addeage(int from,int to)
{
    e[m].to=to;
    e[m].next=head[from];
    head[from]=m++;
}

int dfs(int x,int pre)
{
    int i;
    int t;
    for(i=head[x];i!=-1;i=e[i].next)
    {
        if(pre==e[i].to)
        {
            continue;
        }
        t=e[i].to;
        dfs(t,x);
        dp[x][0]+=max(dp[t][1],dp[t][0]);
        dp[x][1]+=dp[t][0];
    }
    return 0;
}

int main()
{
    int i;
    int n,from,to;
    int ans;
    while(scanf("%d",&n)!=EOF)
    {
        memset(head,-1,sizeof(head));
        memset(dp,0,sizeof(dp));
        for(i=1;i<=n;i++)
        {
            scanf("%d",&dp[i][1]);
        }
        for(i=1;i<=n-1;i++)
        {
            scanf("%d%d",&from,&to);
            addeage(from,to);
            addeage(to,from);
        }
        dfs(1,0);
        ans=max(dp[1][0],dp[1][1]);
        cout<<ans<<endl;
    }
    return 0;
}