leetcode 42.接雨水

給定 n 個非負整數表示每個寬度為 1 的柱子的高度圖，計算按此排列的柱子，下雨之后能接多少雨水。

上面是由數組 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度圖，在這種情況下，可以接 6 個單位的雨水（藍色部分表示雨水）。 感謝 Marcos 貢獻此圖。

示例:

輸入: [0,1,0,2,1,0,1,3,2,1,2,1]
輸出: 6

思路：
　　1：找到和當前能和當前位置形成最大凹槽的位置
　　   有兩種情況
　　　　　　1. 能找到比當前位置高的位置
          2. 找不到比當前高的位置
    2. 計算每個凹槽的容量

int trap(vector<int>& height) {
        int ans = 0, len = height.size();

        for(int i = 0; i < len-1; i++) {
            int j = i+1, max = j, sum = 0;
            //起始位置為0，以及起始位置相鄰位置更高都直接進行下一個， 這些情況下，無法接雨水
            if(height[i] == 0 && height[j] >= height[i]) continue; 
            //找到比當前位置更高或相同高度的位置
            while(j < len && height[j] < height[i]) {
                //存在這樣的情況，當前位置后面不存在比當前位置高的位置
                //那么只能退而求其次，找到當前位置后面最高的位置，這樣
                //才能滿足接的雨水最多
                if(height[max] <= height[j]) max = j;
                //把當前位置和目標位置之間的方塊數加起來
                //當前位置和目標位置之間的面積減去之間的方塊數
                //就是能接的雨水數目
                sum += height[j];
                j++;
            }
            //能找到不比當前位置低的位置
            if(j < len) {
                ans += height[i]*(j-i-1) - sum;
                i = j-1;
            } else if(max != i+1){//找不到比當前位置高的位置
                int sum1 = 0;
                for(int m = i+1; m <= max; m++)  sum1 += height[m];
                ans += (height[max]*(max-i)-sum1);
                i = max-1;
            }
        }
        return ans;
    }