Leetcode刷题java之232用栈来实现队列

一时失言乱红尘 2021-10-30 03:54 216阅读 0赞

执行结果：

通过

显示详情

执行用时 :87 ms, 在所有 Java 提交中击败了69.93%的用户

内存消耗 :34.3 MB, 在所有 Java 提交中击败了47.71%的用户

题目：

使用栈实现队列的下列操作：

push(x) -- 将一个元素放入队列的尾部。  
pop() -- 从队列首部移除元素。  
peek() -- 返回队列首部的元素。  
empty() -- 返回队列是否为空。  
示例:

MyQueue queue = new MyQueue();

queue.push(1);  
queue.push(2);  
queue.peek(); // 返回 1  
queue.pop(); // 返回 1  
queue.empty(); // 返回 false  
说明:

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

思路：

用两个栈来实现队列

入队，就直接入栈就行了

出队的话我们要让先进的先出，我们可以再用一个栈来实现，把栈一出栈的入栈到栈二，就实现了反转。

显示栈顶元素和出队的操作类似

为空就直接判断栈一是否为空就可以了

代码：

class MyQueue {
    
        Stack<Integer> stack1=new Stack<>();
        Stack<Integer> stack2=new Stack<>();
        /** Initialize your data structure here. */
        public MyQueue() {
            
        }
        
        /** Push element x to the back of queue. */
        public void push(int x) {
            stack1.push(x);
        }
        
        /** Removes the element from in front of queue and returns that element. */
        public int pop() {
            while(!stack1.empty())
            {
                stack2.push(stack1.pop());
            }
            int result=stack2.pop();
            while(!stack2.empty())
            {
                stack1.push(stack2.pop());
            }
            return result;
        }
        
        /** Get the front element. */
        public int peek() {
            while(!stack1.empty())
            {
                stack2.push(stack1.pop());
            }
            int result=stack2.peek();
            while(!stack2.empty())
            {
                stack1.push(stack2.pop());
            }
            return result;
        }
        
        /** Returns whether the queue is empty. */
        public boolean empty() {
            return stack1.empty();
        }
    }
    
    /**
     * Your MyQueue object will be instantiated and called as such:
     * MyQueue obj = new MyQueue();
     * obj.push(x);
     * int param_2 = obj.pop();
     * int param_3 = obj.peek();
     * boolean param_4 = obj.empty();
     */