码迷,mamicode.com
首页 > 其他好文 > 详细

[LeetCode] Min Stack

时间:2015-01-08 12:51:24      阅读:114      评论:0      收藏:0      [点我收藏+]

标签:

Design a stack that supports push, pop, top, and retrieving the minimum element in constant time.

  • push(x) -- Push element x onto stack.
  • pop() -- Removes the element on top of the stack.
  • top() -- Get the top element.
  • getMin() -- Retrieve the minimum element in the stack.
class MinStack {
    stack<int> s;
    stack<int> s_min;
    
public:
    void push(int x) {
        s.push(x);
        if (s_min.empty() || x <= s_min.top()) {
            s_min.push(x);
        }
    }

    void pop() {
        if (s.top() == s_min.top()) {
            s_min.pop();
        }
        s.pop();
    }

    int top() {
        return s.top();
    }

    int getMin() {
        return s_min.top();
    }
};

老题了,用两个栈就可以了,后来想了个用链表的,可是爆内存了。唔唔唔。下面是链表的:

struct node {
    int val;
    node *next;
    node *min;
    node():val(-1), next(NULL), min(NULL){}
    node(int v):val(v), next(NULL), min(NULL){}
};

class MinStack {
    node head;
    
public:
    void push(int x) {
        node *top, *pre;
        if(head.next == NULL) {
            head.next = new node(x);
            top = head.next;
            top->min = top;
        } else {
            pre = head.next;
            head.next = new node(x);
            top = head.next;
            top->next = pre;
            if (x <= pre->min->val) {
                top->min = top;
            } else {
                top->min = pre->min;
            }
        }
    }

    void pop() {
        if (head.next == NULL) return;
        node *tmp = head.next;
        head.next = tmp->next;
        delete tmp;
    }

    int top() {
        return head.next->val;
    }

    int getMin() {
        return head.next->min->val;
    }
};

 

[LeetCode] Min Stack

标签:

原文地址:http://www.cnblogs.com/easonliu/p/4210502.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!