前几天抱着试试的态度投了了一个公司,面试的时候让我手撸代码两个栈实现一个队列,我硬是没整出来,太尴尬了,这确实是一个很基础的算法题
不说了,进正题
一.思路分析
首先我们要知道,栈是后进先出,队列是先进先出,怎么实现呢?我们需要用到两个栈s1存数据,s2中转栈。
很多人写的时候,直接将数据压入s1,然后将s1中的数据依次压入s2,然后依次将s2中的数据出栈,这样确实实现了队列先进先出的原理,但再次存入数据时,因为s2中的数据已成定序,在出队列就不能先进先出了,咋整呢?
方法如下:
每次存入数据时,判断s1是否为空,为空则直接存入s1,否则,将s1中的数据依次出栈并存入s2,此时s1为空,再将数据存入s1,在依次将s2中的数据存入s1,这样就保证了每次存入数据都在s1栈底,出队列时依次输出s1中的数据即可。
二.代码
package xianxingbiao;
import java.util.Stack;
public class stacktoqueue {
public static void main(String[] args) {
// TODO Auto-generated method stub
MyQueue<Integer> queue=new MyQueue<>();
queue.push(1);
queue.push(2);
queue.pop();
queue.push(3);
queue.pop();
queue.pop();
}
}
class MyQueue<ElementType>{
private Stack<ElementType> s1=new Stack<ElementType>();
private Stack<ElementType> s2=new Stack<ElementType>();
public void push(ElementType value){
if(s1.isEmpty()){
s1.push(value);
}else{
while(!s1.isEmpty())
s2.push(s1.pop());
s1.push(value);
while(!s2.isEmpty())
s1.push(s2.pop());
}
}
public void pop(){
System.out.print(s1.pop()+" ");
}
}
结果