两个栈实现一个队列(java)

一、问题描述

使用两个栈实现一个队列

二、算法分析

栈是先进后出,因此两个可以模拟实现先进先出

三、算法设计

定义数据结构

Stack<Integer> stack1

Stack<Integer> stack2

对于push操作:元素入队列时,将其加入到stack1中

对于pop操作:元素出队列时,先判断stack2是否为空,若不为空,则从stack2中弹出一个元素;若为空,则将stack1中的所有元素弹出到stack2中,然后弹出一个元素

四、代码实现

import java.util.Stack;

public class Solution {
   Stack<Integer> stack1 = new Stack<Integer>();
   Stack<Integer> stack2 = new Stack<Integer>();
   
   public void push(int node){
	   stack1.push(node);
   }
   public int pop(){
	   if(stack1.isEmpty()&&stack2.isEmpty()){
		   System.out.println("Queue is empty.");
	   }
	   if(stack2.isEmpty()){
		   while(!stack1.isEmpty()){
			   stack2.push(stack1.pop());
		   }
	   }
	   return stack2.pop();
   }
   public static void main(String[] args) {
	   Solution queue = new Solution();
	   queue.push(1);
	   queue.push(2);
	   queue.push(3);
	   System.out.println(queue.pop());
	   queue.push(4);
	   System.out.println(queue.pop());
	   System.out.println(queue.pop());
	   System.out.println(queue.pop());
   }
}


猜你喜欢

转载自blog.csdn.net/tanjie_123/article/details/52996827