题目链接:. - 力扣(LeetCode)
首先我们要明白要形成容器,接雨水
需要右边的柱子高于本身,并且需保证左方有高于本身的柱子
也就是这样,才会形成容器
这道题的解法之一是单调栈,并且需保证是递减的,才能形成容器
下面有代码解析:
typedef struct Stack
{
//int val;
int* arr;
int size;
}ST;
void StackInit(ST* obj )
{
obj->arr = (int*)malloc(sizeof(int)*20005);//创建一个数组,存下标,这个数值于题目给的数组大小有关
obj->size = 0;
}
bool StackIsEmpty( ST* obj )
{
return obj->size == 0 ;
}
int StackTop( ST* obj )
{
return obj->arr[obj->size-1];
}
int StackPop(ST* obj)
{
int i = 0;
i = obj->arr[obj->size-1];
obj->size--;
return i;
}
void StackPush( ST* obj , int pi)
{
obj->arr[obj->size++] = pi;
}
int trap(int* height, int heightSize) {
ST obj;
StackInit(&obj);
int ans = 0;
for( int i = 0 ; i < heightSize ; i++)
{
while( !StackIsEmpty(&obj) && height[StackTop(&obj)] < height[i] )
// 这里是为了保证第一个元素进栈,和当栈里为空时跳出
// 例如:0 1 0 2 第一个元素进栈 然后 1 大于 0 , 0 被弹出 ,栈为空 , 此时不可能形成容器 ,1 就进栈 ,也可以这样写
//在外面让第一个元素进栈,栈为空时,break
{
int k = StackPop(&obj);
while( !StackIsEmpty(&obj) && height[StackTop(&obj)] == height[k] )
{ //这里是为了弹出相同元素 2 1 1 1 5 就要弹出 所有1
StackPop(&obj);
}
if( !StackIsEmpty(&obj) ) // 此时形成容器
{
int top = StackTop(&obj);
ans += (fmin(height[i],height[top]) - height[k])*(i - top - 1 );
} // 高 宽
}
StackPush(&obj,i);
}
return ans;
}