225. Implement Stack using Queues

最新推荐文章于 2024-05-01 20:01:28 发布

原创最新推荐文章于 2024-05-01 20:01:28 发布 · 217 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #java

leetcode 专栏收录该内容

40 篇文章

订阅专栏

本文介绍了一种使用两个队列来模拟栈数据结构的方法。通过将一个队列作为主要队列，另一个作为辅助队列，并在需要弹出元素时将主要队列中除最后一个元素外的所有元素转移至辅助队列，以此实现栈的基本操作。

题目：

Implement the following operations of a stack using queues.

push(x) -- Push element x onto stack.
pop() -- Removes the element on top of the stack.
top() -- Get the top element.
empty() -- Return whether the stack is empty.

Notes:

You must use only standard operations of a queue -- which means only push to back, peek/pop from front, size, and is empty operations are valid.
Depending on your language, queue may not be supported natively. You may simulate a queue by using a list or deque (double-ended queue), as long as you use only standard operations of a queue.
You may assume that all operations are valid (for example, no pop or top operations will be called on an empty stack).

思路：

用两个队列去模拟一个栈，和之前的用两个栈去模拟一个队列很相似，关键在于pop操作的模拟。

我的思路是保证q1是主要的队列，q2是辅助队列。

每次push时都是插入到q1中，当需要pop时，将q1中除了最后一个元素的所有元素remove并push到q2，然后remove最后一个元素，最后再将q1复制成q2。

代码：

class MyStack {
    public List<Integer> q1 = new ArrayList<Integer>();
    public List<Integer> q2 = new ArrayList<Integer>();
    
    // Push element x onto stack.
    public void push(int x) {
        q1.add(x);
    }

    // Removes the element on top of the stack.
    public void pop() {
        if(q1.size() == 0){return;}
        if(q1.size() == 1){
            q1.remove(0);
        }else{
            while(q1.size() > 1){
                q2.add(q1.get(0));
                q1.remove(0);
            }
            q1.remove(0);
            
            q1.addAll(q2);
            q2.clear();
        }
    }

    // Get the top element.
    public int top() {
       if(q1.size() == 1){
            return q1.get(0);
        }else{
            while(q1.size() > 1){
                q2.add(q1.get(0));
                q1.remove(0);
            }
            int value = q1.get(0);
            q2.add(q1.get(0));
            q1.remove(0);
            
            q1.addAll(q2);
            q2.clear();
            return value;
        } 
    }

    // Return whether the stack is empty.
    public boolean empty() {
        return (q1.size()==0 && q2.size()==0);
    }
}

之后看了discuss，看到很多种解法，主要使用List的poll和peek方法，由于我对这两个函数用的不多，因此写出来的代码更难看。。。