每天6道题之第十五题：接雨水（栈）

前言

题目描述

题目解析

接雨水这一题乍一看，没看懂，再看，噢！原来就是说如果某一点左右两边的最大值都比自己大那就说明能够接（不会流出来），但是接多少就需要我们计算了：
接水量=左右两边的最大值中的最小值与当前值的差。
而且要注意开头和结尾的接水量一定为0
就按照这个思路就很好理解这道题，但是如何来做呢？

解题思路

（1）求出高度数组中每个元素的左边最大值，保存在一个数组里
（2）求出高度数组中每个元素的右边最大值，保存在一个数组里
（3）遍历高度数组，接水量=min(leftMax[i],rightMax[i])-height[i].累加即为结果。

trap函数代码

int trap(vector<int> &height){
	int n=height.size();
	if(n==0) return 0;
	vector<int> leftMax(n);
	leftMax[0]=height[0];
	for(int i=1;i<n;i++){
	//将当前元素与前一个元素相比，最大的那个放在leftMax中
		leftMax[i]=max(leftMax[i-1],height[i]);
	}
	vector<int> rightMax(10);
	rightMax[n-1]=height[n-1];
	for(int i=n-2;i>=0;i--){
		rightMax[i]=max(rightMax[i+1],height[i]);
	}
	
	int ans=0;
	for(int i=0;i<n;i++){
		ans+=min(leftMax[i],rightMax[i])-height[i];
	}
	return ans;
}

完整代码

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

class Solution{
	public:
		int trap(vector<int> &height){//传递数组之类的值，一定要传地址，所以会加& 
			int n=height.size();
			if(n == 0) return 0;
			vector<int> leftMax(n);
			leftMax[0]=height[0];//左边没有元素等于自身 
			for(int i=1;i<n;i++){
				leftMax[i]=max(leftMax[i-1],height[i]);
			}
			
			vector<int> rightMax(n);
			rightMax[n-1]=height[n-1];//右边没有元素等于自身 
			for(int i=n-2;i>=0;i--){
				rightMax[i]=max(rightMax[i+1],height[i]);
			} 
			
			int ans=0;
			for(int i=0;i<n;i++){
				ans=ans+min(leftMax[i],rightMax[i])-height[i];
				 
			}
			return ans;
		}
};


int main(){
	int h;
	vector<int> v;
	while(cin>>h){
		v.push_back(h);
	}
	int res=Solution().trap(v);//就是说传进去一个数组，要返回一个雨水量的值
	cout<<res<<endl;
	return 0; 
	
}