小米秋招2019算法题——求下雨时能积多少水

题目

给出一组数组， 表示一组宽为1的小方块的高度。 求下雨时， 可以积水多少

一张很有灵魂的图：

示例：
输入： [0,1,0,2,1,0,1,3,2,1,2,0]
输出： 6

我的答案

function MiTwo(arr){
    //一层一层的思考
    let arr1 = [...new Set(arr.concat([]).sort(function(a,b){return a-b}))]
    let arr2 = []
    arr1.forEach(item => item != 0 && arr2.push(item) )
    let total = 0
/*    while (arr1.length){   //这里是我的第一版， 结果通过率只有67%。  原因就是没考虑到 [0,0,0,2,0,0,0,3,2,0,2,0]， 这种情况。 
        let temp = arr1.shift()
        if(temp != 0){
            let l = arr.indexOf(temp)
            let r = arr.lastIndexOf(temp)

            for(let i = l; i < r; i++){
                if(arr[i] < temp){
                    total++
                }
            }
        }
    }*/
    for(let i = 1; i <= arr2[arr2.length-1]; i++){
        let l = arr.indexOf(arr2[0])
        let r = arr.lastIndexOf(arr2[0])
        for(let j = l; j < r; j++){
            if(arr[j] < arr2[0]){
                total++
            }
        }
        if(i == arr2[0]){
           arr2.shift()
        }
    }
    return total
}
let arr = [0,1,0,2,1,0,1,3,2,1,2,0]
let res = MiTwo(arr)
console.log(res)