用两个队列实现一个栈的功能?要求给出算法和思路:
- 入栈:将元素进队列A
- 出栈:判断队列A中元素的个数是否为1,如果等于1,则出队列(输出),如果不等于1,队列A中的元素依次出队列并放入队列B,直到队列A中的元素留下一个,然后出队列(输出),再把队列B中的元素依次出队列并放回队列A。
import java.util.Stack;
/*
用两个栈实现一个队列的功能?要求给出算法和思路
入队: 将元素进栈A
出队: 判断栈B是否为空, 如果为空,则将栈A中所有元素pop, 并push进栈B中, 栈B出栈一个;
如果不为空,栈B直接出栈一个。
*/
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.empty()) {
while (!stack1.empty()) {
stack2.push(stack1.pop());
}
return stack2.pop();
} else {
return stack2.pop();
}
}
}