题目:
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
思路:
对于每个柱子(不包括最左边和最右边的,因为这两根柱子无论如何也装不下水),找出其左边最高的柱子(高度为a),找出其右边最高的柱子(高度为b),当第i根柱子的高度c小于a和b时,这根柱子可以装Math.min(a,b)-c的水。当c>=a或者c>=b时,这根柱子不能装水。leftDP中的第i个元素表示第i根柱子左边最高柱子的高度,其值可以由第i-1根柱子左边最高柱子的高度和第i-1根柱子的高度去max得到。rightDP 同理。
public int trap(int[] height) {int len = height.length;int[] leftDP = new int[len];int[] rightDP = new int[len];leftDP[0] = 0;rightDP[len-1] = 0;for(int i = 1; i < len-1; i++){leftDP[i] = Math.max(leftDP[i-1], height[i-1]);}for(int i = len-2; i >= 0; i--){rightDP[i] = Math.max(rightDP[i+1], height[i+1]);}int maxTrap = 0;for(int i = 1; i < len-1; i++){if(leftDP[i] > height[i] && rightDP[i] > height[i]){maxTrap += Math.min(leftDP[i], rightDP[i])-height[i];}}return maxTrap;}