算法：接雨水

问题

Given n non-negative integers representing an elevation map where the width of each bar is 1, compute how much water it is able to trap after raining.

样例

接雨水

如上图所示，海拔分别为 [0,1,0,2,1,0,1,3,2,1,2,1], 返回 6.

思路

思路是先看最底下一层能接多少雨水，然后再依次推断到第二层，第三层......最高层。

实现

public static int trapRainWater(int[] heights) {
        // write your code here
        // 先找到最大值
        int max = 0;
        for (int i=0;i<heights.length;i++) {
            max = Math.max(max, heights[i]);
        }
        int rains = 0;
        // 思路是先看最底下一层能接多少雨水，然后再依次推断到最高层
        for (int floor = 0; floor < max; floor++) {
            // 当层数为floor层时
            // 从第1位到第一个非0位之间的0都置为1（便于统计这一层的雨水）
            for (int i=0;i<=heights.length-1;i++) {//统一使用0-start
                if (heights[i] == 0) {
                    heights[i] = 1;
                } else {
                    break;
                }
            }
            // 从最后1位到倒数第一个非0位之间的0都置为1（便于统计这一层的雨水）
            for (int i=heights.length-1;i>=0;i--) {//统一使用0-start
                if (heights[i] == 0) {
                    heights[i] = 1;
                } else {
                    break;
                }
            }
            for (int i=0;i<heights.length;i++) {
                if (heights[i] == 0) {
                    //经过前面的准备工作，我们可以很轻松的分辨出来哪些是雨水
                    rains++;
                } else {
                    //这里要做个操作，就是把数组里的所有元素都减1，为了上面一层的计算做准备
                    heights[i]--;
                }
            }

        }
        return rains;
    }

如果文章对你有帮助的话，不要忘记打赏小编哦