leetcode42. 接雨水（c++）

题目

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

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

思路

有凹陷的地方才能接雨水，即装满雨水后，所有凹陷都被填充，所以只需计算当前面积，在计算装满雨水后的面积，两者相减即可求得雨水的量。
具体做法：
先计算当前面积存在temp中。
设置开始、结束双指针。
我们可以按层地计算，装满雨水后的面积。即找到开始或结束指针所指的值较小的一个，开始到结束这一段长度都乘以较小值，则计算出了较小值所在层装了雨水后的面积。例第一个值为1，第最后一个值为2，类似于木桶原理，其能装载的容量取决于最短板。用tmp记录较大的值
然后迭代找到比tmp大的值，重复计算，直到开始和结束位置相同。
最后返回装满雨水后的面积-装满雨水前的面积temp。

class Solution {
public:
	int trap(vector<int>& height)
	{
		int len = height.size();
		if (len == 0)//过滤特殊情况
			return 0;

		int start = 0;//开始位置
		int end = len - 1;//结束为止
		int temp = 0, sum = 0;
		int tmp = 0;
		for (int i = 0; i<len; ++i)//计算装雨水前的面积
			temp += height[i];

		while (start <= end)//双指针计算装雨水后面积
		{
			if (height[start] == height[end])//两者相同，计算其中一个层数的值
			{
				sum += ((end - start + 1)*(height[start] - tmp));//计算当前层数的值，下面其他情况一样
				tmp = height[start];//记录当前较大值，下面其他情况一样
				while (start <= end && height[start]<=tmp)//start找到比当前较大值大的值
					++start;
				while (start <= end && height[end]<=tmp)//end找到比当前较大值大的值
					--end;
			}
			else if (height[start]>height[end])
			{
				sum += ((end - start + 1)*(height[end] - tmp));
				tmp = height[end];
				while (start <= end && height[end]<=tmp)//start找到比当前较大值大的值
					--end;
			}
			else
			{
				sum += ((end - start + 1)*(height[start] - tmp));
				tmp = height[start];
				while (start <= end && height[start]<=tmp)//end找到比当前较大值大的值
					++start;
			}
		}
		return sum - temp;//结果
	}
};