[LeetCode]Trapping Rain Water,leetcodetrapping

來源:互聯網
上載者:User

[LeetCode]Trapping Rain Water,leetcodetrapping

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.

For example, 
Given [0,1,0,2,1,0,1,3,2,1,2,1], return 6.


The above elevation map is represented by array [0,1,0,2,1,0,1,3,2,1,2,1]. In this case, 6 units of rain water (blue section) are being trapped. Thanks Marcos for contributing this image!
題意:求由上面的數組組成的bar所組成的槽所能夠容納的水量,
思路1:首先找到最大的bar然後分別從左向右計算盛水量,計算時先選定一個非0的bar然後如果後面的bar不比當前大,則依次入棧,直到找到大於當前的,然後對站內的bar依次計算盛水量,最後從右向左重複上面的步驟。
代碼1:

    public int trap1(List<Integer> height) {//兩邊向最長的bar掃面遇見比當前小的就入棧,直到遇到大的就彈出計算盛水量        int sum = 0;        int highIndex = 0;        int max = Integer.MIN_VALUE;        for(int i = 0; i < height.size(); ++ i){            if(height.get(i) > max){                max = height.get(i);                highIndex = i;            }        }        int left;        Stack<Integer> s = new Stack<Integer>();        for(int i = 0; i < highIndex;){            while (i < highIndex && height.get(i) == 0){                i ++;            }            left = height.get(i);            while (i< highIndex) {                while (i < highIndex && height.get(i) <= left) {                    s.push(height.get(i));                    i++;                }                while (!s.isEmpty()) {                    int temp = s.peek();                    s.pop();                    sum += (left - temp);                }                left = height.get(i);                i ++;            }        }        int right = 0;        for(int i = height.size() - 1; i > highIndex;){            while (i > highIndex && height.get(i) == 0){                i --;            }            right = height.get(i);            while (i > highIndex){                while (i > highIndex && height.get(i) <= right){                    s.push(height.get(i));                    i --;                }                while (!s.isEmpty()){                    sum += right - s.peek();                    s.pop();                }                right = height.get(i);                i --;            }        }        return sum;    }
思路2:和上面的計算思路類似,這次不必要首先計算最大的bar,而是每次比較左右兩邊選定的bar的大小,小的那個先計算,依次迴圈。空間O(1),時間O(N)
代碼:
    public int trap(List<Integer> height){        int sum = 0;        int left = 0, right = height.size() - 1;        while (left < right && height.get(left) == 0){            left ++;        }        while (right > left && height.get(right) == 0){            right -- ;        }        int leftHeight,rightHeight;        while (left < right) {            leftHeight = height.get(left);            rightHeight = height.get(right);            if (leftHeight <= rightHeight ){//當左邊bar高度小於右邊的高度的時候,左邊先掃描,直到左邊遇到比leftHeight 高為止                while (left < right && height.get(left) <= leftHeight){                    sum += leftHeight - height.get(left);                    left ++;                }            }else {                while (left < right && height.get(right) <= rightHeight ) {                    sum += rightHeight  - height.get(right);                    right--;                }            }        }        return sum;    }



聯繫我們

該頁面正文內容均來源於網絡整理,並不代表阿里雲官方的觀點,該頁面所提到的產品和服務也與阿里云無關,如果該頁面內容對您造成了困擾,歡迎寫郵件給我們,收到郵件我們將在5個工作日內處理。

如果您發現本社區中有涉嫌抄襲的內容,歡迎發送郵件至: info-contact@alibabacloud.com 進行舉報並提供相關證據,工作人員會在 5 個工作天內聯絡您,一經查實,本站將立刻刪除涉嫌侵權內容。

A Free Trial That Lets You Build Big!

Start building with 50+ products and up to 12 months usage for Elastic Compute Service

  • Sales Support

    1 on 1 presale consultation

  • After-Sales Support

    24/7 Technical Support 6 Free Tickets per Quarter Faster Response

  • Alibaba Cloud offers highly flexible support services tailored to meet your exact needs.