solution.cpp 610 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40
#include <bits/stdc++.h>
using namespace std;
class Solution
{
public:
	int trap(vector<int> &height)
	{
		int res = 0;
		int left = 0, left_max = 0;
		int right = height.size() - 1, right_max = 0;
		while (left < right)
		{
			if (height[left] < height[right])
			{
				if (height[left] > left_max)
				{
					left_max = height[left];
				}
				else
				{
					res += left_max - height[left];
				}
				left++;
			}
			else
			{
				if (height[right] > right_max)
				{
					right_max = height[right];
				}
				else
				{
					res += right_max - height[right];
				}
				right--;
			}
		}
		return res;
	}
};