蓝桥杯 结点选择（树形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的正整数。

状态：dp[u][0]表示不选择结点u的最大权值；

            dp[u][1]表示选择结点u的最大权值；

状态转移方程：dp[u][1]=∑dp[v][0]+pow[v]；   dp[u][0]=∑max(dp[v][0],dp[v][1]);     （求和为u的全部子结点v）

代码：

#include <bits/stdc++.h>
using namespace std;

const int maxV=100005;
vector<int> node[maxV];
int visit[maxV];
int dp[maxV][2];
int Pow[maxV]; 

void dfs(int u)
{
    visit[u]=1;
    int len=node[u].size();
    for(int i=0;i<len;i++)
    {
        int v=node[u][i];
        if(!visit[v])
        {
           dfs(v);
           dp[u][0]+=max(dp[v][0],dp[v][1]);
           dp[u][1]+=dp[v][0];
        }
    }
    dp[u][1]+=Pow[u];
}

int main()
{
    int n,u,v;
    memset(dp,0,sizeof(dp));
    memset(visit,0,sizeof(visit));
    scanf("%d",&n);
    for(int i=1; i<=n; i++)  
        node[i].clear();  
    for(int i=1;i<=n;i++)
       scanf("%d",&Pow[i]);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        node[u].push_back(v);
        node[v].push_back(u);
    }
    dfs(1);
    printf("%d\n",max(dp[1][0],dp[1][1]));
    return 0;
}