用两个栈实现队列

代码

/**
 * 
 * 编写一个类，用两个栈实现队列，支持队列的基本操作(add、poll、peek)。
 * 
 * @author dream
 *
 */
public class Problem02_TwoStacksImplementQueue {

	public static class myQueue{
		
		Stack<Integer> stack1;
		Stack<Integer> stack2;
		
		public myQueue() {
			stack1 = new Stack<Integer>();
			stack2 = new Stack<Integer>();
		}
		/**
		 * add只负责往stack1里面添加数据
		 * @param newNum
		 */
		public void add(Integer newNum){
			stack1.push(newNum);
		}
		
		/**
		 * 这里要注意两点：
		 * 1.stack1要一次性压入stack2
		 * 2.stack2不为空，stack1绝不能向stack2压入数据
		 * @return
		 */
		public Integer poll(){
			if(stack1.isEmpty() && stack2.isEmpty()){
				throw new RuntimeException("Queue is Empty");
			}else if(stack2.isEmpty()){
				while (!stack1.isEmpty()) {
					stack2.push(stack1.pop());
				}
			}
			return stack2.pop();
		}
		
		public Integer peek(){
			if(stack1.isEmpty() && stack2.isEmpty()){
				throw new RuntimeException("Queue is Empty");
			}else if(stack2.isEmpty()){
				while (!stack1.isEmpty()) {
					stack2.push(stack1.pop());
				}
			}
			return stack2.peek();
		}
	}
}

刷题

leetcode-cn.com/problems/yo…

精选题解

直通BAT算法精讲