面试题 用两个栈实现一个队列

那天听朋友说了一个他遇到的面试题：用两个栈实现一个队列。

今天自己实现下。

思路就是栈是先进后出，比如push进去的是1,2,3，那pop出来的就是3,2,1.而队列是先进先出，push的是1,2,3，则pop出的还是1,2,3.因为有2个栈可用，可以把栈1中的数据pop后push到栈2中，再从栈2中pop出来就可以了。

class MyQueue<T>{
	Stack<T> s1 = new Stack<T>();
	Stack<T> s2 = new Stack<T>();
	
	public void push(T t){
		while(s2.size() > 0){//把s2中没有pop的再push回s1
			s1.push(s2.pop());
		}
		s1.push(t);
	}
	
	public T pop(){
		while(s1.size() > 0){
			s2.push(s1.pop());
		}
		return s2.pop();
	}
}

	public static void main(String[] args) {
		
		MyQueue<Integer> q = new MyQueue<>();
		q.push(1);
		q.push(2);
		q.push(3);
		System.out.println(q.pop());
		System.out.println(q.pop());
		q.push(4);
		q.push(5);
		System.out.println(q.pop());
		System.out.println(q.pop());
		System.out.println(q.pop());
	}

输出：

1
2
3
4
5

其实即使只有一个栈也可以实现出一个队列。

class MyQueue<T>{
	Stack<T> s1 = new Stack<T>();
	
	public void push(T t){
		s1.push(t);
	}
	
	public T pop(){
		return s1.remove(0);
	}
}

但前提是该队列有remove（）方法。api中的Stack继承自Vector，Vector中有remove（）。