力扣 225题:用队列实现栈

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

实现 MyStack 类:

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



来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/implement-stack-using-queues
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解法:
queue2做数据最终存储,queue1用于反转队列。
每次先把queue2的旧数据移到queue1中,再把新数据加到queue2中。

class MyStack {

    public MyStack() {

    }
    
    Queue<Integer> queue1 = new ArrayDeque();
    Queue<Integer>queue2 = new ArrayDeque();

    public void push(int x) {
        while (!queue2.isEmpty()){
            queue1.offer(queue2.poll());
        }
        queue2.offer(x);
        while (!queue1.isEmpty()){
            queue2.offer(queue1.poll());
        }
        
    }

    public int pop() {
        return queue2.poll();
    }

    public int top() {
        return queue2.peek();
    }

    public boolean empty() {
        return queue1.isEmpty() && queue2.isEmpty();
    }
}

/**
 * Your MyStack object will be instantiated and called as such:
 * MyStack obj = new MyStack();
 * obj.push(x);
 * int param_2 = obj.pop();
 * int param_3 = obj.top();
 * boolean param_4 = obj.empty();
 */

我有切糕
863 声望45 粉丝

Java架构师,技术经理