1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
| class Solution { public: int trap(vector<int>& height) { int n = height.size(); int res = 0; vector<int> leftMax(n, 0), rightMax(n, 0); for(int i = 1; i < n; i ++) leftMax[i] = max(leftMax[i - 1], height[i - 1]); for(int i = n - 2; i >= 0; i --) rightMax[i] = max(rightMax[i + 1], height[i + 1]);
for(int i = 0; i < n; i ++){ int _min = min(leftMax[i], rightMax[i]); if(_min > height[i]) res += _min - height[i]; } return res; } };
|