剑指offer 21. 包含min函数的栈

版权声明:版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Dby_freedom https://blog.csdn.net/Dby_freedom/article/details/83245936

原题

定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。

Reference Answer

思路分析

思路:使用一个辅助栈保存最小值

# -*- coding:utf-8 -*-
class Solution:
    def __init__(self):
        self.stack = []
        self.min_num =[]
    def push(self, node):
        # write code here
        self.stack.append(node)
        if self.min_num and self.min_num[-1] < node:
            self.min_num.append(self.min_num[-1])
        else:
            self.min_num.append(node)
        
    def pop(self):
        # write code here
        if self.stack:
            self.min_num.pop()
            return self.stack.pop()
        
        return None
    def min(self):
        # write code here
        return self.min_num[-1] if self.min_num else None

C++ Version:

class Solution {
    stack<int> store;
    stack<int> store_min;
public:
    void push(int value) {
        store.push(value);
        if (store_min.empty()){
            store_min.push(value);
        }
        else{
            if (value <= store_min.top()){
                store_min.push(value);
            }
            else{
                store_min.push(store_min.top());
            }
        }
    }
    void pop() {
        store.pop();
        store_min.pop();
    }
    int top() {
        return store.top();
    }
    int min() {
        return store_min.top();
    }
};

Note:

python 中 list.poplist.pop() 是不同的,自己就是误用 pop()pop 一直报错!

没有更多推荐了,返回首页