题目要求如下:
用两个栈实现一个队列。请实现他的两个函数appendTail和deleteHead,分别完成在队列尾部插入节点和在队列头部删除节点的功能。
题目要求要用两个栈实现队列的两个功能:队列尾部插入和队列头部删除。下面分别阐述这两个方面。
尾部插入,即将新的元素插入到队列的尾部,队列具有FIFO即先进先出的特性,而栈具有LIFO特性,因此如果我们想实现队列,只需进行两次入栈出栈操作,即可将LIFO转化为FIFO。因此我们可以先将插入的元素放到第一个栈中,
头部删除,根据上面的思想,我们只需将第一个栈中的元素逐个弹出压到第二个栈里,此时再讲栈顶删除即可。(注意一定要先把第一个栈清空并压第二个栈)。
根据以上原理,可写代码如下:
package com.offer;
import java.util.ArrayDeque;
public class UseStackToMakeQueue {
public static void main(String[] args) {
Deque deque = new Deque();
for (int i = 0; i < 10; i++) {
deque.add(i);
}
System.out.println(deque.get());
}
}
class Deque{
private ArrayDeque<Integer> stack1 = new ArrayDeque<Integer>();
private ArrayDeque<Integer> stack2 = new ArrayDeque<Integer>();
public void add(int n){
//队尾插入节点
stack1.push(n);
}
public int get(){
//队头删除节点
while (!stack1.isEmpty()) {
stack2.push(stack1.pop());
}
if (stack2.isEmpty()) { //已经将stack1的元素全部清空并压入第二个栈,若此时stack2仍为空则队列为空!
throw new RuntimeException("队列为空,无法删除!!");
}
return stack2.pop();
}
}