当前位置：首页 > article >正文

LeetCode 232.用栈实现队列

article 2024/11/15 16:58:45

题目

请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作（push、pop、peek、empty）：

实现 MyQueue 类：

void push(int x) 将元素 x 推到队列的末尾
int pop() 从队列的开头移除并返回元素
int peek() 返回队列开头的元素
boolean empty() 如果队列为空，返回 true ；否则，返回 false
说明：

你只能使用标准的栈操作 —— 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。
你所使用的语言也许不支持栈。你可以使用 list 或者 deque（双端队列）来模拟一个栈，只要是标准的栈操作即可。

在这里插入图片描述
提示：
1 <= x <= 9
最多调用 100 次 push、pop、peek 和 empty
假设所有操作都是有效的（例如，一个空的队列不会调用 pop 或者 peek 操作）

思路

因为队列为先进先出。而栈为先进后出，顺序刚好相反。那么我们只需要准备两个栈，s1和s2。
将数据分别通过栈s1和s2.即可做到先进先出。

但需要注意的是，因为可能连续push多个数，pop一次，然后继续push多个数。那么就形成了连续的段1和连续的段2。我们将段1放入s2后。需要先将此时s2的栈顶返回，才是我们想要pop的数，然后再继续放入段2。如果是将段1和段2都放入s2时再返回s2的栈顶，不会是我们想要pop的数。因为此时s2中只是段间符合队列的顺序，但段与段之间不符合队列的顺序。

同时我们也因为push的不连续性和push时先都将数只缓存在s1的操作，只有在查询队列队头或者弹出队头时才涉及s2的操作，导致s1和s2都是有可能为空的。我们要判断整个队列是否为空，需要看s1和s2是否同时为空。
①s1为空，s2不为空的情况：s1的数已经全部赋给s2了。此时s2就是队列。
②s2为空，s1不为空的情况：正在连续push数中，s1此时为队列的逆序顺序。

代码

class MyQueue {
    Stack<Integer> s1 = new Stack<Integer>();
    Stack<Integer> s2 = new Stack<Integer>();

    public MyQueue() {

    }
    
    public void push(int x) {
        s1.push(x);
    }
    
    public int pop() {
        if(s2.empty()){
            while(!s1.empty()){
            s2.push(s1.peek());
            s1.pop();
        }
        }
        int ans = (int)s2.peek();
        s2.pop();
        return ans;
    }
    
    public int peek() {
        if(s2.empty()){
        while(!s1.empty()){
            s2.push(s1.peek());
            s1.pop();
        }}
        return s2.peek();
    }
    
    public boolean empty() {
        return s1.empty()&&s2.empty();
    }
}


 */