[leetcode]Trapping Rain Water

Trapping Rain Water

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.

difficulty:hard

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!

题目写的很清楚，图也很直观，就是计算这些树状图在下雨的时候能存多少雨水。

我用的方法是从左到右遍历每个数，找到每个坑并计算存水的数量。当数是降序的时候，说明是坑的左边，数是升序的时候就是坑的右边。遍历的时候如果遇到坑左就记录一下位置，如果遇到坑右就开始计算能存的水，并把已经计算过的坑填上，最后重新记录可以作为坑左的位置。

int trap(vector<int>& height) {
	vector<int> left;
	int result = 0;
	int i;
	for (i=0; i< height.size(); i++){//找到第一个不为0的数
		if (height[i] > 0)
			break;
	}
	left.push_back(i);//记录不为0的位置
	i++;
	for (; i < height.size(); i++){//开始遍历

		if (height[i]>height[i - 1]){//升序情况
			int max = left[left.size() - 1];

			for (int j = left.size() - 1; j >= 0; j--){
				if (i - j > 1){//坑左与坑右距离要大于1
					if (height[left[j]] >= height[i]){//往前找到高度大于等于当前坑右的坑左

						for (int k = left[j] + 1; k < i; k++){//记录坑并填坑
							result += height[i] - height[k];
							height[k] = height[i];
						}

						left.push_back(i);//当前坑右往后当做坑左来计算
						break;
					}
				}
				if (height[left[j]] > height[max])//记录坑右中最高的
					max = left[j];

			}

			if (height[max] < height[i]){//前面没找到比当前坑右高的坑左，则用坑左里最高的那个
				if (i - max>1){//坑左与坑右距离要大于1
					for (int k = max + 1; k < i; k++){//记录坑并填坑
						result += height[max] - height[k];
						height[k] = height[max];
					}
					left.push_back(i);//当前坑右往后当做坑左来计算
				}
				else{
					left.push_back(i);//当前坑右往后当做坑左来计算
				}

			}

		}
		else{//降序或者不变则记录为坑左
			left.push_back(i);

		}
	}
	return result;

}