使用两个栈来模拟队列

栈

package stack_queue;

/**
 * 使用链表来实现栈
 * @author User
 *
 */
public class MyStack {
	Node top;
	
	/**
	 * 压栈
	 * @param data
	 */
	public void push(int data) {
		// 先把传入的数字构造成一个节点，因为是链表嘛！
		Node node = new Node(data);
		if(top == null) {
			top = node;
		}else {
			// 直接将新节点作为栈顶元素
			node.next = top;
			top = node;
		}
	}
	
	
	/**
	 * 出栈
	 * @return
	 */
	public int pop() {
		if(isEmpty()) {
			throw new RuntimeException("栈是空的!");
		}
		int data = top.data;
		top = top.next;
		return data;
	}
	
	
	/**
	 * 返回栈顶元素
	 * @return
	 */
	public int peek() {
		if(isEmpty()) {
			throw new RuntimeException("栈是空的!");
		}
		return top.data;
	}
	
	
	/**
	 * 栈中元素总数
	 * @return
	 */
	public int length() {
		if(top == null) {
			return 0;
		}
		Node temp = top;
		// 计数器
		int count = 0;
		while(temp != null) {
			temp = temp.next;
			count++;
		}
		
		return count;
	}
	
	
	/**
	 * 判空
	 * @return
	 */
	public boolean isEmpty() {
		return top == null;
	}
}


class Node{
	int data;
	Node next;
	
	public Node(int data) {
		this.data = data;
	}
}

队列

package stack_queue;
/**
 * 使用两个栈来模拟队列的入队出队操作
 * @author User
 *
 */
public class MyQueue {
	// 定义两个栈来模拟队列
	MyStack stackA = new MyStack();// 用于入队
	MyStack stackB = new MyStack();// 用于出队
	
	/**
	 * 入队
	 * @param data
	 */
	public void add(int data) {
		stackA.push(data);
	}
	
	/**
	 * 出队
	 * @return
	 */
	public int remove() {
		// 这里很关键，只有stackB是空的的时候才往里面压入数据，不然的话，如果stackB中如果还有数据，这时候突然入队了一个新的就马上从stackA压到stackB里面，这就乱了
		if(stackB.isEmpty()) {
			while(!stackA.isEmpty()) {
				stackB.push(stackA.pop());
			}
		}
		
		int data = stackB.pop();
		
		return data;
	}
	
	
	
}

测试类

package stack_queue;

public class Test {
	public static void main(String[] args) {
		MyQueue queue = new MyQueue();
		
		queue.add(1);
		queue.add(2);
		queue.add(3);
		queue.add(4);
		queue.add(5);
		
		System.out.println(queue.remove());
		System.out.println(queue.remove());
		System.out.println(queue.remove());
		System.out.println(queue.remove());
		// 出队过程中突然入队
		queue.add(6);
		System.out.println(queue.remove());
		System.out.println(queue.remove());
	}
}	

输出