时间限制:C/C++ 1秒,其他语言2秒 空间限制:C/C++ 32M,其他语言64M 热度指数:623773
题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
1、分析
定义两个栈s1和s2,其中s1作为入队的主战场,s2作为出队的主战场。元素入队时,直接让元素入s1;元素出队时,由于队列遵循先进先出的元素,而先进的元素处在栈s1的栈底,不能优先出来,所以把栈s1中的所有元素逐个出栈s1并入栈s2,当s1中所有元素都出栈并入栈到s2中时候,s2栈顶元素就是最先入队的元素,出队(s2的栈顶出栈)即可。
算法过程:
(1)入队:元素入s1
(2)出队:先判断s2中是否有元素,如果没有,把s1中所有元素都出栈并入栈到s2中(也要注意判断s1是否为空),当把s1中元素都出栈并入栈s2后,对s2执行出栈操作实际上就是执行出队操作。
2、代码
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(!stack2.isEmpty()){
return stack2.pop();
}
while(!stack1.isEmpty()){
stack2.push(stack1.pop());
}
return stack2.pop();
}
}