LeetCode225:用队列实现栈

发布于:2024-03-23 ⋅ 阅读:(69) ⋅ 点赞:(0)

题目描述
请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通栈的全部四种操作(push、top、pop 和 empty)。

实现 MyStack 类:

void push(int x) 将元素 x 压入栈顶。
int pop() 移除并返回栈顶元素。
int top() 返回栈顶元素。
boolean empty() 如果栈是空的,返回 true ;否则,返回 false 。

示例:
输入:
[“MyStack”, “push”, “push”, “top”, “pop”, “empty”]
[[], [1], [2], [], [], []]
输出:
[null, null, null, 2, 2, false]

解释:
MyStack myStack = new MyStack();
myStack.push(1);
myStack.push(2);
myStack.top(); // 返回 2
myStack.pop(); // 返回 2
myStack.empty(); // 返回 False

解题思想
用两个队列que1和que2实现队列的功能,que2其实完全就是一个备份的作用,把que1最后面的元素以外的元素都备份到que2,然后弹出最后面的元素,再把其他元素从que2导回que1。

代码

class MyStack {
public:
    queue<int> qu1;
    queue<int> qu2;

    MyStack() {

    }

    void push(int x) {
        qu1.push(x);
    }

    int pop() {
        //留下qu1队列中的最后一个元素
        while (qu1.size() != 1) {
            qu2.push(qu1.front());
            qu1.pop();
        }
        int result = qu1.front();
        qu1.pop();
        
        //再将qu2的元素全放到qu1中
        qu1 = qu2;
        while (!qu2.empty()) {
            qu2.pop();
        }
        
        return result;
        
    }

    int top() {
        return qu1.back();
    }

    bool empty() {
        return qu1.empty();
    }
};

解题思想2
使用一个队列就能实现,pop的时候将最后一个元素之前的元素全部重新入队。

代码

class MyStack {
public:
    queue<int> qu1;


    MyStack() {

    }

    void push(int x) {
        qu1.push(x);
    }

    int pop() {
        int size = qu1.size();
        while (size!=1) {
            qu1.push(qu1.front());
            qu1.pop();
            size--;
        }
        int result = qu1.front();
        qu1.pop();

        return result;

    }

    int top() {
        return qu1.back();
    }

    bool empty() {
        return qu1.empty();
    }
};
本文含有隐藏内容,请 开通VIP 后查看

网站公告

今日签到

点亮在社区的每一天
去签到