当前位置:   article > 正文

接雨水(C语言)

接雨水(C语言)

题目链接:. - 力扣(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;

}

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/凡人多烦事01/article/detail/347664
推荐阅读
相关标签
  

闽ICP备14008679号