225. 用队列实现栈(简单)

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

push(x) – 元素 x 入栈
pop() – 移除栈顶元素
top() – 获取栈顶元素
empty() – 返回栈是否为空

注意:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/implement-stack-using-queues

思路:

思路类似232的,但这里使用一个队列即可,因为每次push元素时,我们就是给队列add一个,我们循环size-1次,每次add一个remove即可

代码:

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
class MyStack {
//类似232栈实现队列,但用一个队列来完成即可,每次push一个,循环remove再add,就可以了
Queue<Integer> a;
// Queue<Integer> b;
/** Initialize your data structure here. */
public MyStack() {
a = new LinkedList<Integer>();
}

/** Push element x onto stack. */
public void push(int x) {
a.add(x);
for(int i=0; i<a.size()-1; i++)
a.add(a.remove());
}

/** Removes the element on top of the stack and returns that element. */
public int pop() {
return a.remove();
}

/** Get the top element. */
public int top() {
return a.peek();
}

/** Returns whether the stack is empty. */
public boolean empty() {
return a.isEmpty();
}
}
/**
* Your MyStack object will be instantiated and called as such:
* MyStack obj = new MyStack();
* obj.push(x);
* int param_2 = obj.pop();
* int param_3 = obj.top();
* boolean param_4 = obj.empty();
*/