【题目】:编写一个类,用两个栈实现队列,支持队列的基本操作(add、poll、peek)
【思考】:栈的特点是先进后出,队列的特点是先进先出。因此我们要想用两个栈来完成队列就必须考虑如何将其转换成先进先出。逆向想一下,将两个栈反过来刚好可以实现队列的操作。
如果还不太理解的话,我们来一起看一下这张图来理解一下。
【方案】:具体实现是把一个栈作为压入栈,记为stackPush;一个栈作为弹出栈,记为stackPop。因为数据压入栈时,顺序是先进后出,因此我们只需要将stackPush栈的数据再压入到stackPop栈中,顺序就变回来了。这样子stackPush压入时顺序和stackPop弹出时顺序就相同了。
代码:
import java.util.Queue;
import java.util.Stack;
public class StackQueue {
public Stack<Integer> stackPush;
public Stack<Integer> stackPop;
public StackQueue() {
stackPush = new Stack<>();
stackPop = new Stack<>();
}
/**压栈,将数据压入stackPush栈中*/
public void add(int pushNum) {
stackPush.push(pushNum);
}
//移除并返回队列头部的元素,若果队列为空,返回null;
public int poll() {
if(stackPush.empty() && stackPop.empty()) {
System.out.println("该队列为空");
}else if(stackPop.empty()){
while(!stackPush.empty()) {
/**若弹出栈为空,且压入栈不为空时,将压入栈中元素进行弹栈并且压入到弹出栈中*/
stackPop.push(stackPush.pop());
}
}
return stackPop.pop();
}
//返回队列头部的元素,如果队列为空,则返回null;
public int peek() {
if(stackPush.empty() && stackPop.empty()) {
System.out.println("该队列为空");
}else if(stackPop.empty()){
while(!stackPush.empty()) {
/**若弹出栈为空,且压入栈不为空时,将压入栈中元素进行弹栈并且压入到弹出栈中*/
stackPop.push(stackPush.pop());
}
}
return stackPop.peek();
}
public static void main(String[] args) {
StackQueue stack = new StackQueue();
stack.add(1);
stack.add(2);
stack.add(3);
stack.add(4);
stack.add(5);
stack.add(6);
System.out.println(stack.poll());
System.out.println(stack.peek());
}
}