232. 用栈实现队列

使用栈实现队列的下列操作:

push(x) – 将一个元素放入队列的尾部。 pop() – 从队列首部移除元素。 peek() – 返回队列首部的元素。 empty() – 返回队列是否为空。 示例:

MyQueue queue = new MyQueue();

queue.push(1); queue.push(2);
queue.peek(); // 返回 1 queue.pop(); // 返回 1 queue.empty(); // 返回 false 说明:

你只能使用标准的栈操作 – 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。 你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。 假设所有操作都是有效的 (例如,一个空的队列不会调用 pop 或者 peek 操作)。

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/implement-queue-using-stacks 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题解:

栈后进先出,队列先进先出,用栈实现队列

1,需要两个栈inStack 和 outStack,

2,入队时inStack 压栈。

3,出队时如果outStack不为空,则outStack 出栈, 如果outStack为空, 把inStack中的元素,依次pop并压入outStack中,inStack元素全部压入outStack中后,返回outStack pop。

4,peek时,与出队列类似,pop换做peek即可。

5,isEmpty 判断inStack 和 outStack都为空才是空

代码如下:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
public class _232_用栈实现队列{

	private Stack<Integer> inStack;
	private Stack<Integer> outStack;

	public _232_用栈实现队列(){
		inStack = new Stack<>();
		outStack = new Stack<>();
	}

	public void push(int x){
		inStack.push(x);
	}

	public int pop(){
		if (!outStack.isEmpty()) {
			return outStack.pop();
		}else {
			while(!inStack.isEmpty()){
				outStack.push(inStack.pop());
			}

			return outStack.pop();
		}
	}

	public int peek(){
		if (!outStack.isEmpty()) {
			return outStack.peek();
		}else{
			while(!inStack.isEmpty()){
				outStack.push(inStack.pop());
			}
			return outStack.peek();
		}
	}

	public boolean empty(){
		return inStack.isEmpty() && outStack.isEmpty();
	}
}