小米秋招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)