蓝桥杯 —— 结点选择 —— 树状DP

问题描述

有一棵 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的题目。印尼为前几天做一道比较简单的树状DP的题目。那是道题是codeforces 274B。。

那道题是说将这棵树上面的子树权值+1 或者 -1.最终使得所有权值全部为0.求最少的操作次数是多少。

刚开始我用那道题的答题模板敲了一遍，但是求出的最大值不对。在主函数中遍历所有的顶点。。dfs（v，l）。

初始的那个v是取得，立一个flag标记这个顶点，然后它的子节点就应该取到。

但是仔细想一想，不一定是交叉选择的，v取的话，子节点肯定不能取。

但是如果v结点不取的时候呢，他的子节点就出现了两种情况了，取或者不取。

所以这道题应该用一个DP的知识解决。

首先，用前向星的方法将树中所有的结点连在一起。这样从第一个结点开始dfs。如果他的下一个结点的num = -1；就跳出搜索。

最后的答案为ans = max(dp[1][1],dp[1][0]);

注：dp[i][0]代表第i个结点不要。dp[i][1]代表取到第i个结点。。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <string>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <map>
#include <set>
#include <sstream>
#include <queue>
#include <stack>
#define INF 0x3f3f3f3f
#define mem(a,b) memset(a,b,sizeof(a));
#define For(a,b) for(int i = a;i<b;i++)
#define ll long long
#define MAX_N 100010

using namespace std;

struct qxx
{
    int v;
    int next;
}edge[200010];
int head[MAX_N],dp[MAX_N][3];
int num;

void add(int v,int u)
{
    edge[num].v = u;
    edge[num].next = head[v];
    head[v] = num ++;

    edge[num].v = v;
    edge[num].next = head[u];
    head[u] = num ++;
    return ;
}

void dfs(int v,int l)
{
    for(int i = head[v]; i != -1; i = edge[i].next)
    {
        int u = edge[i].v;
        if(u == l) continue;
        dfs(u,v);
        dp[v][0] += max(dp[u][0],dp[u][1]);
        dp[v][1] += dp[u][0];
    }
    return ;
}

int main()
{
    int n;
    scanf("%d",&n);
    int ans = 0;
    num = 0;
    mem(dp,0);
    mem(head,-1);
    for(int i = 1; i<=n; i++)
    {
        scanf("%d",&dp[i][1]);
    }
    for(int i = 0; i<n-1; i++)
    {
        int a,b;
        scanf("%d %d",&a,&b);
        add(a,b);
    }
    dfs(1,-1);
    ans = max(dp[1][1],dp[1][0]);
    printf("%d\n",ans);
    return 0;
}