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

leetcode[155] Min Stack

时间:2014-12-18 18:32:14      阅读:145      评论:0      收藏:0      [点我收藏+]

标签:style   blog   color   sp   div   log   bs   amp   as   

实现一个堆,可以push,pop,top,和getMin

主要的应该是getMin,如何找到最小的数。如果每次要找最小的数遍历一边,那貌似没什么意义了。

做法是,用两个堆,一个用来正常的操作,另一个是记录到当前最小值,如果来一个数字,比最小值的那个对的top还小那就,两个堆都要push,如果pop的时候pop的数和最小值的那个堆的top一样,那就最小值的那个堆也要pop。在getMin的时候,只要返回最小值那个堆的top就可以了。

稍微模拟一下几个数字进出堆就知道了。

class MinStack {
public:
    stack<int> sta, minsta;
    void push(int x) {
        sta.push(x);
        if (minsta.empty() || x <= minsta.top())
            minsta.push(x);
    }

    void pop() {
        if (!minsta.empty() && minsta.top() == sta.top())
            minsta.pop();
        sta.pop();
    }

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

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

 

leetcode[155] Min Stack

标签:style   blog   color   sp   div   log   bs   amp   as   

原文地址:http://www.cnblogs.com/higerzhang/p/4172289.html

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