网易2017春招[编程题]堆砖块@Java

链接：https://www.nowcoder.com/questionTerminal/040924ba0e64423b8a3fe2f75a56934a
来源：牛客网

[编程题]堆砖块

• 热度指数：5778时间限制：1秒空间限制：32768K
• 算法知识视频讲解

小易有n块砖块，每一块砖块有一个高度。小易希望利用这些砖块堆砌两座相同高度的塔。为了让问题简单，砖块堆砌就是简单的高度相加，某一块砖只能使用在一座塔中一次。小易现在让能够堆砌出来的两座塔的高度尽量高，小易能否完成呢。

输入描述:

输入包括两行： 第一行为整数n(1 ≤ n ≤ 50)，即一共有n块砖块 第二行为n个整数，表示每一块砖块的高度height[i] (1 ≤ height[i] ≤ 500000)

输出描述:

如果小易能堆砌出两座高度相同的塔，输出最高能拼凑的高度，如果不能则输出-1. 保证答案不大于500000。

示例1

输入

3 2 3 5

输出

5

package go.jacob.day915;

import java.util.Scanner;

public class Demo5 {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int[] height = new int[n];
		int sum = 0;
		for (int i = 0; i < n; i++) {
			height[i] = sc.nextInt();
			sum += height[i];
		}
		// dp0与dp1都表示低塔高度，dp0为上一轮，dp1为当前轮次
		// 长度设为2*sum+1，是因为高度差的范围是-sum到sum
		int[] dp0 = new int[2 * sum + 1];
		int[] dp1 = new int[2 * sum + 1];
		// 初始化
		for (int i = 0; i <= 2 * sum; i++)
			dp0[i] = -1;// 初始化为-1表示不存在该情况
		dp0[sum] = 0;

		for (int i = 0; i < n; i++) {
			int high = height[i]; 
			// 把j定义为（B堆高度-A堆高度）+sum; //
			// 范围从[-sum,sum]变为[0,sum*2];加上sum是为了方便用数组表示
			for (int j = 0; j <= 2 * sum; j++) {
				dp1[j] = dp0[j]; 
				// 放在A堆上
				if (j >= high && dp0[j - high] != -1) {
					dp1[j] = Math.max(dp1[j], dp0[j - height[i]]);
				} 
				// 放在B堆上，B堆高度增大
				//注意，这里的if一定要有两个判断条件
				//（如果初始值设为Integer.MIN_VALUE可以只有一个判断条件）
				if (j + high <= 2 * sum && dp0[j + high] != -1) {
					dp1[j] = Math.max(dp1[j], dp0[j + high]+high);
				}
			}
			//交换数组，保证每一轮循环dp0为上一轮
			int[] s = dp0;
			dp0 = dp1;
			dp1 = s;
		}

		
		if (dp0[sum] == 0)
			System.out.println(-1);
		else
			System.out.println(dp0[sum]);
		sc.close();

	}
}