【剑指offer】用两个栈实现队列的入队出队操作

剑指offer

题目要求

用两个栈来实现一个队列,完成队列的Push(入栈)和Pop(出栈)操作。

队列中的元素为int类型

补充

栈的特性:先进后出(LIFO, Last In First Out),类似于水桶。
队列的特性:先进先出(FIFO, FIFO In First Out),类似于水管。

核心思想

利用两个栈,一个用于入队,一个用于出队,入队时,直接放入栈1,出队时,由栈1倒到栈2,再出(将先出栈的又变成先入栈的,后出栈的变成后入栈的,因此达到队列的要求,通俗的讲,就是负负得正)

完整代码如下

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() {
		//如果栈2是空,就把栈1内容弹出来入栈2,如果上述操作后仍然为空,则返回-1;
		if(stack2.empty()){
	//这里,只有在stack2为空时,即pop完之后,才可以插入stack1的元素。否则会发生错误
			while(stack1.empty() != true){
				stack2.push(stack1.pop());
			}
		}
		if(stack2.empty()){
			//即stack1也为空
			return -1;
		}
		//否则,可以出队(出栈)
		return stack2.pop();
		
	}

	
	public static void main(String[] args) {
		  Solution sl = new Solution();
		  sl.push(6);
		  sl.push(7);
		  sl.push(8);
		  
		  System.out.println(sl.pop());
		  System.out.println(sl.pop());

	}
}

猜你喜欢

转载自blog.csdn.net/weixin_42967016/article/details/84561396