剑指offer面试题7——用两个栈实现队列

package offer7;

import common.Stack;

/**
* 用两个栈实现一个队列
* @author chibozhou
*/
public class Queue<T> {
//队列当前大小
private int curSize;
//队列最大容量
private int maxSize;
//两个栈
private Stack<T> stack1;
private Stack<T> stack2;

/**
* 初始化队列
* @param maxSize 设置最大容量
*/
public Queue(int maxSize){
//健壮性判断
if(maxSize<=0){
System.out.println("队列大小不能小于1");
return;
}

//初始化栈
stack1 = new Stack<T>(maxSize);
stack2 = new Stack<T>(maxSize);
this.maxSize = maxSize;
}

/**
* 入队
* @param obj 入队的元素
*/
public boolean push(T obj){
//健壮性判断
if(obj==null){
System.out.println("入队元素不能为空！");
return false;
}

//队已满
if(stack1.getCurSize()>=maxSize){
System.out.println("队以满！");
return false;
}

//入队
stack1.push(obj);
curSize++;
return true;
}

/**
* 出队
*/
public T pop(){
//对为空
if(curSize==0){
System.out.println("队为空！");
return null;
}

//将stack1中的所有元素转移到stack2中，再从stack2栈顶取元素即可
while(stack1.getCurSize()>0){
stack2.push(stack1.pop());
}
return stack2.pop();
}

/**
* 打印队列
*/
@Override
public String toString() {
System.out.println("Queue---maxSize:"+maxSize+"---curSize:"+curSize+"------");
//队列为空
if(curSize==0){
System.out.println("队列为空");
return null;
}

//若数据都在stack1中，则转到stack2中
if(stack1.getCurSize()>=0){
while(stack1.getCurSize()>0){
stack2.push(stack1.pop());
}
}

//打印stack2中的元素
for(int i=0;i<stack2.getCurSize();i++){
System.out.println(stack2.top());
}
System.out.println("Queue--------------------------------------------------");
return null;
}

}