2020-05-12 LeetCode 115 最小棧

題目:

 設計一個支持 pushpoptop 操作,並能在常數時間內檢索到最小元素的棧。

  • 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.

提示:

poptopgetMin 操作總是在 非空棧 上調用。

 比較基礎的數據結構題,相比常規棧就是多了一個最小元素的相關操作,最開始的時候想法市用兩個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不等
1

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章