题目:
设计一个支持 push
,pop
,top
操作,并能在常数时间内检索到最小元素的栈。
push(x)
—— 将元素 x 推入栈中。pop()
—— 删除栈顶的元素。top()
—— 获取栈顶元素。getMin()
—— 检索栈中的最小元素。
示例:
输入:
["MinStack","push","push","push","getMin","pop","top","getMin"]
[[],[-2],[0],[-3],[],[],[],[]]
输出:
[null,null,null,null,-3,null,0,-2]
解释:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.
提示:
pop
、top
和 getMin
操作总是在 非空栈 上调用。
比较基础的数据结构题,相比常规栈就是多了一个最小元素的相关操作,最开始的时候想法市用两个int,一个储存数据一个作为标志位来表示最小值(只有数据的话最小值的初始化会有问题),复现也比较快。需要注意的点是栈溢出的处理情况,有一个测试用例一直push null进去,开始没考虑这个,后面是改成了对于多余的元素选择不push跳过,栈的最大内存设为800对于其他用例就不会出错了。后面查看题解发现用的是另外以一个栈来辅助储存最小值,思路本质上是没区别的,就是后者看起来更规范一点,所以参考其把代码修改了一下,如下:
#define STACK_MAX 800
typedef struct {
int top;
int data[STACK_MAX];
} Stack;
typedef struct {
Stack AllData;
Stack MinData;
} MinStack;
MinStack *minStackCreate() {
MinStack *stack = malloc(sizeof(MinStack));
stack->AllData.top = -1;
stack->MinData.top = -1;
return stack;
}
void minStackPush(MinStack *obj, int x) {
if (obj->AllData.top == (STACK_MAX - 1))
return;
obj->AllData.data[++obj->AllData.top] = x;
if (obj->MinData.top == -1 || x <= obj->MinData.data[obj->MinData.top]) {
obj->MinData.data[++obj->MinData.top] = x;
}
}
void minStackPop(MinStack *obj) {
if (obj->AllData.top == -1)
return;
if (obj->AllData.data[obj->AllData.top] == obj->MinData.data[obj->MinData.top]) {
obj->MinData.top--;
}
obj->AllData.top--;
}
int minStackTop(MinStack *obj) {
return obj->AllData.data[obj->AllData.top];
}
int minStackGetMin(MinStack *obj) {
return obj->MinData.data[obj->MinData.top];
}
void minStackFree(MinStack* obj) {
free(obj);
}
执行结果内存基本都是12M左右,用时波动比较大,32ms-44ms不等