剑指Offer || ：栈与队列（简单）

系列文章目录

1. 【栈与队列】

---

---

前言

这个专栏记录leetcode刷题记录，以剑指Offer第2版为主

---

1. 用两个栈模拟队列

原题链接：剑指 Offer 09. 用两个栈实现队列

题目：
用两个栈实现一个队列。队列的声明如下，请实现它的两个函数 appendTail 和 deleteHead ，分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素，deleteHead 操作返回 -1 )

示例：

输入：
[“CQueue”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[3],[],[]]
输出：[null,null,3,-1]

1 <= values <= 10000
最多会对 appendTail、deleteHead 进行 10000 次调用

代码

class CQueue {
    
private:
    stack<int> s1;
    stack<int> s2;
    
    void push_In_S2() {
    
        while (!s1.empty()) {
    
            s2.push(s1.top());
            s1.pop();
        }
    }

public:
    CQueue() {
    }
    // 入队列操作，入栈s1
    void appendTail(int value) {
    
        s1.push(value);
    }
    
    // 出栈操作
    int deleteHead() {
    
        if (s2.empty()) {
    
            if (s1.empty()) {
    
                return -1;
            }
            push_In_S2(); // 该函数负责把s1的元素入栈到s2
        }

        int value = s2.top();
        s2.pop();
        return value;
    }
};

/** * Your CQueue object will be instantiated and called as such: * CQueue* obj = new CQueue(); * obj->appendTail(value); * int param_2 = obj->deleteHead(); */

2. 包含min函数的栈

题目：
定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中，调用 min、push 及 pop 的时间复杂度都是 O(1)。

原题链接：包含min函数的栈

代码：

可借助辅助栈，每次入栈都记录最小值，让最小值保持在辅助栈的最上面。

class MinStack {
    
private:
    stack<int> stk;
    stack<int> min_stk;
public:
    /** initialize your data structure here. */
    MinStack() {
    
        min_stk.push(INT_MAX);
    }
    
    void push(int x) {
    
        stk.push(x);
        min_stk.push(std::min(min_stk.top(), x));
    }
    
    void pop() {
    
        stk.pop();
        min_stk.pop();
    }
    
    int top() {
    
        return stk.top();
    }
    
    int min() {
    
        return min_stk.top();
    }
};