陷阱雨水:

来源:互联网 发布:linux后台运行命令 编辑:程序博客网 时间:2024/06/12 01:15

给定n代表高度图的非负整数,其中每个小节的宽度为1,计算下雨后能捕捉多少水。
例如,给定[0,1,0,2,1,0,1,3,2,1,2,1],返回6。
这里写图片描述
解题思想:
扫描一遍,找到最高的柱子,这个柱子将数组分为两半;
处理左边一半;
处理右边一半
在其一半之中定义一个水位,每次利用水位差来计算该位置所存的水滴数量,最后water+=1,计算总和。
代码:
package ACM;

public class Solution {
public static int trap(int[] A) {
int n = A.length;
int height = 0;
for (int i = 0; i < n; i++)
if (A[i] > A[height]) height = i;

    int water = 0;    for (int i = 0, index = 0; i < height; i++) {        if (A[i] > index) index = A[i];        else water += index - A[i];    }    for (int i = n - 1, index = 0; i > height; i--) {        if (A[i] > index) index = A[i];        else water += index - A[i];    }    return water;}public static void main(String[] args) {    int[] A={0,1,0,2,1,0,1,3,2,1,2,1};    System.out.println(trap(A));}

};

原创粉丝点击