蓝桥杯 算法训练 结点选择 （树形动态规划）

问题描述

有一棵 n 个节点的树，树上每个节点都有一个正整数权值。如果一个点被选择了，那么在树上和它相邻的点都不能被选择。求选出的点的权值和最大是多少？

输入格式

第一行包含一个整数 n 。
接下来的一行包含 n 个正整数，第 i 个正整数代表点 i 的权值。
接下来一共 n-1 行，每行描述树上的一条边。

输出格式

输出一个整数，代表选出的点的权值和的最大值。

样例输入

5
1 2 3 4 5
1 2
1 3
2 4
2 5

样例输出

12

解题思路

value[] 存储每个节点的价值，
dp[i][0]表示包含节点i的子树最大值，dp[i][1]表示不包含节点i的子树最大值，
则动态转移方程：
dp[u][0]+=dp[v][0]，
dp[u][1]+=dp[v][0]+dp[v][1] ,v为u 的子节点.

代码实现

#include <bits/stdc++.h>
#define IO ios::sync_with_stdio(false);\
    cin.tie(0);\
    cout.tie(0);
using namespace std;
#define maxn 100007
int value[maxn];
int dp[maxn][2];
bool mark[maxn];    //标记节点是否使用过
struct node
{
    int to;
    int next;
} edge[maxn*2];
int head[maxn],tot;
int n,m;
void addedge(int u,int v)  //邻接表建树
{
    edge[tot].to = v;
    edge[tot].next = head[u];
    head[u] = tot++;
}
void dfs(int no)
{
    dp[no][0]=value[no];
    dp[no][1]=0;
    for(int i=head[no];i!=-1;i=edge[i].next)
    {
        int to=edge[i].to;
        if(!mark[to])
        {
            mark[to]=1;
            dfs(to);
            dp[no][0]+=dp[to][1];///
            dp[no][1]+=max(dp[to][0],dp[to][1]);
            mark[to]=0;
        }
    }
}
int main()
{
    IO;
    cin>>n;
    memset(head,-1,sizeof head);
    memset(dp,-1,sizeof(dp));
    for(int i=1;i<=n;i++)
        cin>>value[i];
    tot = 0;
    for(int i=1; i<n; i++)
    {
        int u,v;
        cin>>u>>v;
        addedge(u,v);
        addedge(v,u);
    }
    mark[1]=1;
    dfs(1);
    cout<<max(dp[1][0],dp[1][1])<<endl;
    return 0;
}