欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

用两个栈实现队列

程序员文章站 2024-03-18 12:08:10
...
题目描述:

用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。

思路:

两个栈,有两个端口,那么肯定一个是用来入队的,另一个用来出队的。同时,由于栈是先进后出的,那么经过两次的入栈则会变为先进先出,即,第一次先进后出,第二次后进先出,两个加起来就变成了先进先出。

算法实现
public class TwoStacksQueue {
	public Stack<Integer> stackPush;
	public Stack<Integer> stackPop;
	public TwoStacksQueue() {
		stackPush = new Stack<Integer>();
		stackPop = new Stack<Integer>();
	}
	public void add(int pushInt) {
		stackPush.push(pushInt);
	}
	
	public int poll() {
		if(stackPop.empty()&&stackPush.empty()) {
			throw new RuntimeException("Queue is empty");
			//若stackPop为空,则将stackPush中的元素全部移动到栈stackPop中
		}else if(stackPop.empty())
		{
			while(!stackPop.empty()) {
				stackPop.push(stackPush.pop());
			}
		}
		//pop stackPop的栈顶元素
		return stackPop.pop();
	}
	public int peek() {
		if(stackPop.empty()&&stackPush.empty()) {
			throw new RuntimeException("Queue is empty!");
			//若stackPop为空,则将stackPush中的元素全部移动到栈stackPop中
		}else if(stackPop.empty()) {
			while(!stackPush.empty()) {
				stackPop.push(stackPush.pop());
			}
		}
		//输出stackPop的栈顶元素
		return stackPop.peek();
	}
}