剑指Offer-用两个栈实现队列

题目

使用两个栈实现一个队列的入队和出队。

分析

用两个栈实现队列的入队和出队操作,栈的特点是先进后出,队列是先进先出。当从队列中添加数据的时候向stack1添加,出队的时候,从stack2出栈,这时候有三种情况;
第一种:stack1和stack2都为空,这是队列中已经没有值,可以抛出异常或者返回默认值。
第二种:stack1和stack2都不为空,这时候直接从stack2中出栈,也就是栈顶元素。
第三种:当stack2为空的时候,将stack1所有元素都出栈,放到stack2中,这时候原本在stack1中栈底的元素就变成了stack2 的栈顶元素,因此出栈就遵从先入先出原则了。

时间复杂度:push操作O(1),pop也是O(1);

代码实现

import java.util.Stack;


public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    public void push(int node) {
        stack1.push(node);
        
    }
    
    public int pop() {
        if(stack1.empty()&&stack2.empty()){
            throw new RuntimeException("Queue is empty!");
            
        }else if(stack2.empty()){
            while(!stack1.empty()){
                stack2.push(stack1.pop());
            }
        }
        return stack2.pop();
    }
}

以上有理解不到位的地方欢迎指出。

原文地址:https://www.cnblogs.com/dataoblogs/p/14121846.html