LeetCode 232.用栈实现队列

题目

请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(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(双端队列)来模拟一个栈,只要是标准的栈操作即可。

LeetCode 232.用栈实现队列_第1张图片
提示:
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();
    }
}


 */

效率

0ms,击败100.00%使用 Java 的用户。应该不用再优化了。

你可能感兴趣的:(LeetCode,leetcode,java,算法,后端)