LeetCode-232. 用栈实现队列

清疚 2022-09-15 15:59 146阅读 0赞

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

实现 MyQueue 类：

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

思路:之前在剑指offer的时候写过这题,今天写这题,又出现几个需要注意的点。  
1.代码规范,构造方法之外定义引用,进入了构造方法之后初始化对象  
2.特地将stack1出栈的元素添加到stack2当中单独写成一个方法,每每我们写pop或者peek方法的时候,都去判断stack2中是否为空,再去调用写好的方法  
3.pop和peek是有区别的,pop出栈的时候删除栈顶元素,peek取栈顶元素,栈中并没有删除该元素

class MyQueue { 
        Stack<Integer> stack1;
        Stack<Integer> stack2;
        public MyQueue() { 
            stack1=new Stack();
            stack2=new Stack();
        }
        
        public void push(int x) { 
            stack1.push(x);     
        }
        
        public int pop() { 
            if(stack2.empty()){ 
                Stack1PushStack2();
            }
            return stack2.pop();
        }
        
        public int peek() { 
            if(stack2.empty()){ 
                Stack1PushStack2();
            }
            return stack2.peek();
        }
        
        public boolean empty() { 
            return stack1.empty() && stack2.empty();
        }
        public void Stack1PushStack2(){ 
            while(!stack1.empty()){ 
                stack2.push(stack1.pop());
            }
        }
    }
    
    /**
     * 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();
     */