如何使用栈模拟一个队列

Posted 2021-01-14 EasilyAi

队列先进先出。

栈先进后出。

最近看到一道面试题，要求用两个栈模拟一个队列，这个问题在一年前遇到过，记录一下。

使用栈模拟队列，一个肯定不行，首先想到用两个栈来模拟队列。

当队列有数据要入队时，我们同时将这个数据入栈A

此时我们认为，第一次向队列中插入数据完成了。则我们将A中的数据弹出，使用B来接收这些值。

就这样将A中的元素全部弹出并且压进B中，这时就会发现，他们顺序逆转，并且如同队列一样，先进的元素到了栈顶，可以先出。

我们让1出队之后，此时我们还想继续向队列中添加数据。

 

 

此时，2依然是栈B放在栈顶的元素。此时有新的元素加入，我们依然采取刚才的方法，在新数据入队时，插入数据到栈A中。

 插入栈A之后，我们可以对栈B进行判断，如果栈B中还存有元素，则这名刚才添加的元素尚未全部出队，他们仍然遵守先进先出的原则。

当栈B中的元素全部弹出之后，我们就可以将栈A中的元素再次弹出并压进栈B，此时又实现了先进先出。

代码实现：

1. private Stack<Integer> stackA = new Stack<Integer>();

2. private Stack<Integer> stackB = new Stack<Integer>();

3.  

5. /**

6. * 入队操作

7. * @param element 入队的元素

8. */

9. public void enQueue(int element) {

10. stackA.push(element);

11. }

12.  

13. /**

14. * 出队操作

15. */

16. public Integer deQueue() {

17. if(stackB.isEmpty()){

18. if(stackA.isEmpty()){

19. return null;

20. }

21. transfer();

22. }

23. return stackB.pop();

24. }

25. ``

26. ``

27. /**

28. * 栈A元素转移到栈B

29. */

30. private void transfer(){

31. while (!stackA.isEmpty()){

32. stackB.push(stackA.pop());

33. }

34. }

35. ``

36. ``

37. public static void main(String[] args) throws Exception {

38. StackQueue stackQueue = new StackQueue();

39. stackQueue.enQueue(1);

40. stackQueue.enQueue(2);

41. stackQueue.enQueue(3);

42. System.out.println(stackQueue.deQueue());

43. System.out.println(stackQueue.deQueue());

44. stackQueue.enQueue(4);

45. System.out.println(stackQueue.deQueue());

46. System.out.println(stackQueue.deQueue());

48. }