剑指Offer-用两个栈实现队列

最近一直在复习一些算法及数据结构方面的东西,就找了一个适合找工作笔试的题目,在剑指Offer上刷了几道题目,发现对复习知识点还是很有用的,做到重建二叉树这块。递归传值出了点问题,debug半小时才找出错误,所有还是写篇博客记录一下。也推荐要找工作的伙伴去剑指Offer刷题。

       

题目描述

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

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(stack2.size() > 0) {
	    	   return stack2.pop();	
	    	}else {
	    		while(stack1.size() > 0 ) {
	    			stack2.push(stack1.pop());
	    		}
	    		return stack2.pop();
	    	}
	    }
}

很简单,一个栈专门负责进队的操作,假如要出队的话,就将负责进队的这个栈全部出栈到另外一个栈,假如又有进队的元素,直接进入stack1,等stack2的长度为0之后才将stack1出栈到stack2

点赞