那天听朋友说了一个他遇到的面试题:用两个栈实现一个队列。
今天自己实现下。
思路就是栈是先进后出,比如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()。