如何用两个栈来实现一个队列,完成队列的Push和Pop操作。

如何用两个栈来实现一个队列,完成队列的Push和Pop操作(队列中的元素为int类型)

import java.util.Stack;
public class Solution {
    
    
    //创建两个栈stack1、stack2
    Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
	//入队
    public void push(int node) {
    
    
        stack1.push(node);//设定每一次入队都放入stack1中
}
	//出队
    public int pop() {
    
    
        int num = 0;//定义一个num用于存放出队的数
        if (stack1.empty()) return 0;
           while (stack1.size() != 1) {
    
    
           	   //将stack1中除了栈底元素之外元素放入stack2中
               stack2.push(stack1.pop());
           }
           num = stack1.pop();//将栈底元素弹出
           while(!stack2.empty()) {
    
    
               //将stack2中所有元素放回到stack1中
               stack1.push(stack2.pop());
           } 
        return num;
    }
}

思路:

  • 有两个栈,设置其中一个栈stack1为专门存放数字的,另一个栈stack2作为过渡栈;
  • 每次入队时都往stack1中放入,出队时先将stack1中除了栈底最后一个元素都先将其放入stack2中(栈是先进后出,队列是先进先出);
  • 执行完该次循环后(stack1中只剩最后一个栈底元素),将栈底元素弹出并保存,但不要忘了再次利用循环将stack2中移入的元素再次弹出到stack1中,这样就确保了元素的顺序;
  • 最后返回保存了的栈底元素值即可。

猜你喜欢

转载自blog.csdn.net/change_xzt/article/details/115332175