232.用栈实现队列225. 用队列实现栈
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 操作)。
思路
定义两个栈来模拟队列,元素入栈存入A栈,元素出栈存入B栈,并从B栈输出
push(x) – 将一个元素放入队列的尾部。 元素入A栈
pop() – 从队列首部移除元素。 如果B栈不为空,那么移除B栈栈顶元素。如果B栈为空,那么将A栈所有元素移除,并按照移除顺序依次存入B栈,最后移除B栈栈顶元素
peek() – 返回队列首部的元素。 如果B栈不为空,那么移除B栈栈顶元素。如果B栈为空,那么将A栈所有元素移除,并按照移除顺序依次存入B栈,最后输出B栈栈顶元素
empty() – 返回队列是否为空。 如果A和B栈都为空,那么队列为空
代码如下
public class MyQueue {
Stack<Integer> a;
Stack<Integer> b;
public MyQueue() {
a = new Stack<>();
b = new Stack<>();
}
public void push(int x) {
a.push(x);
}
public int pop() {
if(!b.empty()){
return b.pop();
}else {
while(!a.empty()){
b.push(a.pop());
}
return b.pop();
}
}
public int peek() {
if(!b.empty()){
return b.peek();
}else {
while(!a.empty()){
b.push(a.pop());
}
return b.peek();
}
}
public boolean empty() {
if(b.empty() && a.empty()){
return true;
}
return false;
}
}
225. 用队列实现栈
使用队列实现栈的下列操作:
- push(x) – 元素 x 入栈
- pop() – 移除栈顶元素
- top() – 获取栈顶元素
- empty() – 返回栈是否为空
注意:
- 你只能使用队列的基本操作-- 也就是 push to back, peek/pop from front, size, 和 is empty 这些操作是合法的。
- 你所使用的语言也许不支持队列。 你可以使用 list 或者 deque(双端队列)来模拟一个队列 , 只要是标准的队列操作即可。
- 你可以假设所有操作都是有效的(例如, 对一个空的栈不会调用 pop 或者 top 操作)。
思路
用一个队列即可模拟栈。
push(x) – 元素 x 入栈 。元素入队
pop() – 移除栈顶元素。队头到队尾(除了最后一个元素)出队后重新入队,移除队顶元素
top() – 获取栈顶元素 队头到队尾(除了最后一个元素)出队后重新入队,获取队顶元素
empty() – 返回栈是否为空 。判断队列是否为空
代码如下
class MyStack {
Queue<Integer> queue;
public MyStack() {
queue = new LinkedList<>();
}
public void push(int x) {
queue.add(x);
}
public int pop() {
int num = queue.size();
while (num > 1) {
queue.add(queue.poll());
num--;
}
return queue.poll();
}
public int top() {
int num = queue.size();
while (num > 1) {
queue.add(queue.poll());
num--;
}
int result = queue.peek();
queue.add(queue.poll());
return result;
}
public boolean empty() {
if(queue.isEmpty()){
return true;
}
return false;
}
}
问题
top() – 获取栈顶元素.获取栈顶元素时,我光顾着获取元素了,但是却改变了队列的结构。如果只获取一次栈顶元素,我的算法没有问题,但是如果重复在获取一次栈顶元素,就不是现在的值了,需要恢复原来的结构
public int top() {
int num = queue.size();
while (num > 1) {
queue.add(queue.poll());
num--;
}
return queue.peek();
}
改变后
public int top() {
int num = queue.size();
while (num > 1) {
queue.add(queue.poll());
num--;
}
int result = queue.peek();
queue.add(queue.poll());
return result;
}