题目
使用两个栈实现一个队列的入队和出队。
分析
用两个栈实现队列的入队和出队操作,栈的特点是先进后出,队列是先进先出。当从队列中添加数据的时候向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();
}
}
以上有理解不到位的地方欢迎指出。