11.接雨水(LeetCode42)

问题描述 :
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。

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

可使用以下代码，完成其中的trap函数，其中形参height为上述的柱子高度数组，返回接到的雨水量。

输入说明 :
输入若干个非负整数，以空格分隔。

输出说明 :
输出一个整数，表示结果。

输入范例 :
0 1 0 2 1 0 1 3 2 1 2 1

输出范例 :
6

#include<iostream>
#include<vector>
#include<stack>
#include<algorithm> 
using namespace std;

class Solution {
public:
	/*
	单调递减栈
	积水只在低洼处形成，当后面的柱子比前面的低时，是无法接雨水的
	所以使用单调递减栈存储可能储水的柱子，当找到一根比前面高的柱子，就可以计算接到的雨水了 
	时间复杂度O(n),空间复杂度O(n) 
	*/ 
	int trap(vector<int>& height)
	{
		int res = 0;
		stack<int> stk;
		for(int i = 0; i < height.size(); i++) {
			while(!stk.empty() && height[stk.top()] < height[i]) {
				int cur_index = stk.top();
				stk.pop();
				if(stk.empty()) { //栈为空，说明前面没有柱子或者更高的柱子了 
					break;
				}
				int left_index = stk.top();
				int right_index = i;
				int h = min(height[left_index],height[right_index]) - height[cur_index];
				res = res + (right_index - left_index -1) * h; 
			}
			stk.push(i);
		}
	   	return res;
	}
	/*
	暴力法：
	 对于数组中的每个元素，找出下雨后水能达到的最高位置，等于两边最大高度的较小值减去当前高度的值 
	*/ 
	int trap2(vector<int>& height) {
		int ans = 0;
		for(int i = 1; i < height.size()-1; i++) {
			int max_left = 0, max_right = 0;
			for(int j = i; j >= 0; j--) {	//计算当前元素的左边最大的高度(包含当前元素)
				max_left = max(max_left,height[j]); 
			}	
			for(int j = i; j < height.size(); j++) {		计算当前元素的右边最大的高度(包含当前元素)
				max_right = max(max_right,height[j]);
			}
			ans += min(max_left,max_right) - height[i];	//计算这个位置能存储的水量 
		}
		return ans; 
	}
};

int main()
{
    int h;
    vector<int> v;
    while(cin>>h)
    {
        v.push_back(h);
    }
    int res=Solution().trap2(v);
    cout<<res<<endl;
    return 0;
}