单调栈解木板倒水问题

循环操作每个木板,维护一个从栈底到栈顶单调递减的单调栈。

具体算法如下:先输入第i块木板的高度,然后标记下模板编号,记录到变量temp里。接着,temp依次和栈顶元素a比较,如果a的高度小于等于temp的高度,则弹出。

根据单调栈的性质,元素a出栈表明我们已经找到元素a右侧第一个比它大的元素了(这里指的就是temp),元素temp和元素a之间隔的元素个数等于temp的编号减去a的编号再减1,累加结果后我们进行新的比较,重复上述操作,知道栈顶元素的高度大于temp的高度,然后我们再把temp加入栈里。

循环结束后,我们还需要判断栈是否为空,如果不为空,则依次弹出栈顶元素,操作和上面的一样,最后输出结果,算法结束。

#include <stdio.h>
#include <stdlib.h>

#define ERROR 0
#define OK 1

typedef struct Node {
    int id, height;
} Node;

typedef struct Stack {
    Node *elements;
    int max_size, top_index;
} Stack;

void init(Stack *s, int length) {
    s->elements = (Node *)malloc(sizeof(Node) * length);
    s->max_size = length;
    s->top_index = -1;
}

int push(Stack *s, Node element) {
    if (s->top_index >= s->max_size - 1) {
        return ERROR;
    }
    s->top_index++;
    s->elements[s->top_index] = element;
    return OK;
}

int pop(Stack *s) {
    if (s->top_index < 0) {
        return ERROR;
    }
    s->top_index--;
    return OK;
}

Node top(Stack *s) {
    return s->elements[s->top_index];
}

int empty(Stack *s) {
    if (s->top_index < 0) {
        return 1;
    } else {
        return 0;
    }
}

void clear(Stack *s) {
    free(s->elements);
    free(s);
}

int main() {
    int n, ans = 0;
    scanf("%d", &n);
    Stack *stack = (Stack *)malloc(sizeof(Stack));
    init(stack, n);
    Node temp;
    for (int i = 1; i <= n; i++) {
        scanf("%d", &temp.height);
        temp.id = i;
        while (!empty(stack) && top(stack).height <= temp.height) {
            ans = ans + i - top(stack).id - 1;
            pop(stack);
        }
        push(stack, temp);
    }
    while (!empty(stack)) {
        ans = ans + n + 1 - top(stack).id - 1;
        pop(stack);
    }
    printf("%d\n", ans);
    clear(stack);
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

yitahutu79

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值