题目:用两个栈实现一个队列。队列的声明如下,请实现它的两个函数appendTail和deleteHead,分别完成在队列尾部插入节点和在队列头部删除节点的功能。
思路:队列是先进先出的,栈是先进后出的,用stack1来存放每次新增的元素,删除元素就在stack2删除,不过每次删除前要判断stack2中是否没有元素,有元素就删除栈顶的,没有元素就将stack1的元素移动到stack2中。例如,stack1存储1,2,3,现在要删除元素,stack1中栈顶元素为3,不合理,所以将stack1的元素全部移动到stack2,就变成3,2,1,这时候栈顶元素就变成1了,可以正常删除,删除1后,如果stack1存入4,然后再次删除,就要看stack2中是否为空了,此时stack2为3,2,如果将stack1的4移动过来,就变成3,2,4,栈顶为4,显然不合理,所以应该等3,2都删完再将stack1的元素移动过来。
解法:
package question9;
import java.util.Stack;
public class SimpleQueue<T> {
private Stack<T> stack1=new Stack<>();
private Stack<T> stack2=new Stack<>();
public T appendTail(T t){
return stack1.push(t);
}
public T deleteHead(){
if(stack2.size()<=0){
while(stack1.size()>0){
stack2.push(stack1.pop());
}
}
if(stack2.size()==0){
System.out.println("当前队列为空");
return null;
}
return stack2.pop();
}
public static void main(String[] args) {
SimpleQueue<Integer> queue=new SimpleQueue<>();
queue.appendTail(1);
queue.appendTail(2);
queue.appendTail(3);
queue.appendTail(4);
System.out.println(queue.appendTail(5));
System.out.println(queue.deleteHead());
System.out.println(queue.deleteHead());
System.out.println(queue.deleteHead());
System.out.println(queue.deleteHead());
}
}