AC代码
时间复杂度为O(n),空间复杂度为O(1)。只用了i,j,k三个变量作为指针。
思路是:设置两个指针i,j,i指向左边的第一个高大于0的柱子,j指向右边第一个高大于0的柱子。然后,选择这两课柱子中较矮的那个(若两者相等,则选择i),从这个位置开始直到其后(或者其前)的第一个大于它的高度的柱子之间,统计这段区间内的面积。
如这里第一次循环时选择i,则i向右移动到第二个黑色柱子处停下(此时i=3),统计从i=1到i=3之间的距离为1。
上图所示是第一次循环开始时,i,j的位置,以及第一次循环结束时i的位置i’
第二次循环时选择j,j向左移动到第二个黑色柱子处并统计j和j‘之间的面积(为0)。
class Solution {
public:
int trap(vector<int>& height) {
int i = 0,j = height.size()-1;
while(i<height.size() && height[i]==0){
i++;
}
while(j>=0 && height[j] ==0){
j--;
}
int area = 0;
while(i<j){
if(height[i]<=height[j]){
int k = i + 1;
while(k<j && height[k]<=height[i]){
area += height[i] - height[k];
k++;
}
i = k;
}else{
int k = j - 1;
while(i<k && height[k]<height[j]){
area += height[j] - height[k];
k--;
}
j = k;
}
}
return area;
}
};