一、 用栈实现队列
力扣第232题
题目描述:
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push
、pop
、peek
、empty
):
实现 MyQueue
类:
void push(int x)
将元素 x 推到队列的末尾int pop()
从队列的开头移除并返回元素int peek()
返回队列开头的元素boolean empty()
如果队列为空,返回true
;否则,返回false
说明:
- 你 只能 使用标准的栈操作 —— 也就是只有
push to top
,peek/pop from top
,size
, 和is empty
操作是合法的。 - 你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。
示例:
输入: ["MyQueue", "push", "push", "peek", "pop", "empty"] [[], [1], [2], [], [], []] 输出: [null, null, null, 1, 1, false] 解释: MyQueue myQueue = new MyQueue(); myQueue.push(1); // queue is: [1] myQueue.push(2); // queue is: [1, 2] (leftmost is front of the queue) myQueue.peek(); // return 1 myQueue.pop(); // return 1, queue is [2] myQueue.empty(); // return false
解题思路:
push操作:首先将栈 stack1 的元素全部输入到栈 stack2 ,然后将新元素压入栈 stack1 中,最后再将栈 stack2 中的元素重新压入栈 stack1,这样就实现了新加的元素一直处于 stack1 栈底,而 stack1 栈顶元素则是最开始的添加的元素
pop操作:弹出 stack1 的栈顶元素
pop操作:输出 stack1 的栈顶元素(不弹出)
empty操作:判断 stack1 是否为空
代码如下:
class MyQueue {
Stack<Integer> stack1;
Stack<Integer> stack2;
public MyQueue() {
stack1 = new Stack<>();
stack2 = new Stack<>();
}
public void push(int x) {
while(!stack1.isEmpty()) {
int num = stack1.pop();
stack2.push(num);
}
stack1.push(x);
while(!stack2.isEmpty()) {
int num = stack2.pop();
stack1.push(num);
}
}
public int pop() {
if(!stack1.isEmpty()) {
return stack1.pop();
}
return -1;
}
public int peek() {
if(!stack1.isEmpty()) {
return stack1.peek();
}
return -1;
}
public boolean empty() {
return stack1.isEmpty();
}
}
时间复杂度:push操作为O(n),每次循环两次栈,其余操作为O(1)
二、用队列实现栈
力扣第2225题
题目描述:
请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通栈的全部四种操作(push
、top
、pop
和 empty
)。
实现 MyStack
类:
void push(int x)
将元素 x 压入栈顶。int pop()
移除并返回栈顶元素。int top()
返回栈顶元素。boolean empty()
如果栈是空的,返回true
;否则,返回false
。
注意:
- 你只能使用队列的标准操作 —— 也就是
push to back
、peek/pop from front
、size
和is empty
这些操作。 - 你所使用的语言也许不支持队列。 你可以使用 list (列表)或者 deque(双端队列)来模拟一个队列 , 只要是标准的队列操作即可。
示例:
输入: ["MyStack", "push", "push", "top", "pop", "empty"] [[], [1], [2], [], [], []] 输出: [null, null, null, 2, 2, false] 解释: MyStack myStack = new MyStack(); myStack.push(1); myStack.push(2); myStack.top(); // 返回 2 myStack.pop(); // 返回 2 myStack.empty(); // 返回 False
解题思路:
push操作:将新元素进队,然后循环将队头元素取出并插入到队尾,直到队头是新元素为止
其余操作十分简单
代码如下:
class MyStack {
Queue<Integer> queue;
public MyStack() {
queue = new LinkedList<>();
}
public void push(int x) {
queue.offer(x);
int size = queue.size();
while(size > 1) {
queue.offer(queue.poll());
size--;
}
}
public int pop() {
return queue.poll();
}
public int top() {
return queue.peek();
}
public boolean empty() {
return queue.isEmpty();
}
}
时间复杂度:push操作是O(n),其余操作复杂度是o(1)