ALGO-4 结点选择 — 树型动态规划（java）

ALGO—4 结点选择

问题描述

有一棵 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的正整数。

题目分析：
这是一道树型动态规划与dfs深度搜索相结合的题目。主要的思路在于建树和状态转移方程的寻找。

算法分析：
首先是建树。仔细分析输入数据，发现描述的是树上的一条边，而没有指明父子关系，所以需要分情况。详见代码注释。
其次是状态转移方程，逻辑很简单，当取x时，不能取它的孩子x_son；当不取x时，比较x_son取和不取时的值的大小，取两者间的最大值，由于题目最终要求输出是求和的值，所以状态方程为：
dp[x][1] += dp[x_son][0];
dp[x][0] += Math.max(dp[x_son][0], dp[x_son][1]);
另外需要注意的是，建树的过程由于是无向的，所以在dfs和使用状态转移方程之前，我们必须确认x_son是x的孩子。

算法设计：（java）

import java.util.*;

public class Main {
	static int Tree[][];
	static int dp[][];

	public static void BuildTree(int x, int y) {
        //由于输入的数据描述的是树上的一条边，没有指明父子关系，所以分情况
        //如果y是x的孩子
		int i = 0; 
		while (Tree[x][i] != 0) //一直循环，直到Tree[x][i] = 0,也就是直到没有孩子为止
			i++;
		Tree[x][i] = y;
        //如果x是y的孩子
		int j = 0;
		while (Tree[y][j] != 0) //一直循环，直到Tree[y][j] = 0,也就是直到没有孩子为止
			j++;
		Tree[y][j] = x;
	}

	public static void dfs(int x, int x_father) {
		int x_son = Tree[x][0];
		for(int i=0;x_son!=0;i++){
			x_son = Tree[x][i];
			if (x_son != x_father) { //BuildTree时是无向的，此处if条件判断用来确定x_son是x的孩子
				dfs(x_son, x);
                //状态转移方程：
				dp[x][1] += dp[x_son][0];
				dp[x][0] += Math.max(dp[x_son][0], dp[x_son][1]);
			}
		}
	}

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		dp = new int[n+1][2];
		Tree = new int[n*2][n*2];
		for (int i = 1; i <= n; i++) {
			dp[i][1] = sc.nextInt(); //树的第一层为根节点
		}
		for (int i = 0; i < n - 1; i++) {
			int x = sc.nextInt();
			int y = sc.nextInt();
			BuildTree(x, y);
		}
		dfs(1, 0);
		System.out.println(Math.max(dp[1][1], dp[1][0])); //dfs最终状态是：根节点取或不取的最大值
		sc.close();
	}
}

算法思路参考：http://www.cnblogs.com/liuzhen1995/p/6550442.html

附：该java解法在蓝桥OJ上判为运行超时，改为C++后AC了，查阅资料后发现java的效率确实是比C慢的，主要存在的是输入流的速率问题，以下链接已更新了java的再优化AC版，欢迎阅读ALGO-4 结点选择 — 树型动态规划（java再优化版）。