題目:
設計一個支持 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不等