蓝桥杯java节点选择_蓝桥杯节点选择(java)第一道树形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，刚开始表示很难理解，当时主要的疑惑点有一下几点：

1：这个树怎么解决？这么乱的一棵树感觉根本无法下手，因为你不知道那个是根节点，那个是子节点。输出的结果又和那个有关系？

2：树形怎么dp？以前都是遇到线性区间的dp，树形dp，怎么查找连续点，就算找到连续的点交叉点如何处理？

看了一些博客和文章之后，有了解到dp处理树形的特殊手段：

一： 一般来说，树形dp的每个节点都有一个选择性，本题就是该店选择和不选择，dp[i][0]表示不选择，dp[i][1]表示选择该节点。

二：对于树形dp，一般要和搜索结合(更确切的说是递归)结合，对于树的划分层次，一般是选择一个点，然后从这个点进行往下搜索 ，他的邻居都变成他的子节点。也就是相当于从这个根节点除法，可能有很多分叉。也就是有很多分叉会到根节点。但是这有和dp有啥关系呢？dp是从尾到头还是从头到尾？

三：对于顺序的选择肯定是从尾部到顶部，因为dp要的是一个整合结果而不是分散求最值或者其他。那么还有问题就是那么多的跟节点，那么多合的点，还有反向路径不好记录，记录的难度和开销都超级大。如果从尾部推到dp还是有一定难度。

四：我们要先抛开整体看局部这个点，对于某个单点来说(如果他是头节点)，分析到他的结果，如果取他那么他的儿子们都不能取，那么dp[i][1]=dp[儿子们][0] value[i];如果不取他，那么对于每个儿子来说要给最大的，那么dp[i][0]=max(dp[每个儿子][0],dp[每个儿子][1]);这样就得到递推式dp[x][0]=sum(max(dp[num][0],dp[num][1]));dp[x][1]=sum(dp[num][0]) value[i].

五：对于头节点(可以任选)，满足④，对于头节点的儿子也同样满足④，但是唯一不同的是要过滤头节点(准确的说是父节点)。防止死循环。递归搜索的过程是双向的过程，先去再回，我们可以同过先去构造树并且获得后面的一些信息，然后根据后来得到的值进行操作当前等级的dp。

总的来说这个树形dp就是分析某个点的状态方程，通过递归搜索进行划分树。获得结果。并且这题只有n-1条路径，n个点，不用考虑去重。

附上代码：

package 算法训练;

import java.io.BufferedReader;

import java.io.IOException;

import java.io.InputStreamReader;

import java.io.OutputStreamWriter;

import java.io.PrintWriter;

import java.io.StreamTokenizer;

import java.util.ArrayList;

import java.util.List;

public class 节点选择 { // static boolean jud[]; static int dp[][];//dp数组 static int value[];//保存权值 static List []list;//邻接表储存图。节省空间

public static void main(String[] args) throws IOException {

// TODO 自动生成的方法存根

StreamTokenizer in=new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));

PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));

in.nextToken();

int n=(int)in.nval;

value=new int[n];//权值

dp=new int[n][2];

list=new ArrayList[n]; for(int i=0;i();}

for(int i=0;i

{ in.nextToken(); value[i]=(int)in.nval;

}

for(int i=0;i

{ in.nextToken();int t1=(int)in.nval; in.nextToken();int t2=(int)in.nval; list[t1-1].add(t2-1);//添加路径 list[t2-1].add(t1-1); }

dfs(0,-1);//理论上任意n之内节点都可以，但是右侧第一个理论上保证不是这个点的父亲

int value=max(dp[0][0], dp[0][1]);

out.println(value);

out.flush();

}

private static void dfs(int x, int y) {//当前节点，父亲节点

for(int i=0;i

{ int num=list[x].get(i); if(num!=y)//不是父亲节点 { dfs(num,x); dp[x][0]+=max(dp[num][0],dp[num][1]); dp[x][1]+=dp[num][0]; }

}

dp[x][1]+=value[x];//加上自己的权值 }

private static int max(int i, int j) {

// TODO 自动生成的方法存根

return i>=j?i:j;

}

}1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

但是结果只能过7个，

看了下其他人没优化输入的只能过5个超时。我用测试数据测试了一下原因是栈内存溢出。苦逼的Java。。

也可能是因为我比较菜，，想不出好的方法，呵呵，。欢迎大佬踢场。。

文章来源: bigsai.blog.csdn.net，作者：Big sai，版权归原作者所有，如需转载，请联系作者。

原文链接：bigsai.blog.csdn.net/article/details/83627755