使用队列实现栈的下列操作:
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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:利用两个队列实现堆栈:
1.始终保持一个队列为空,另外一个队列被使用;
2.如果a队列被使用时,这时执行pop或者peak操作,那么就循环把a队列的元素都移入b队列中,并定义一个变量存最后一个操作的元素(实现堆栈的pop与peak操作),如果时pop操作,则不存最后一个元素于b队列中,peak则存;
3.如果b队列使用时,同上相反。
代码如下:
class MyStack {
private Queue<Integer> queue1;
private Queue<Integer> queue2;
/** Initialize your data structure here. */
public MyStack() {
queue1 = new LinkedList<Integer>();
queue2 = new LinkedList<Integer>();
}
//始终存一个队列中,另外一个保持空的状态时刻等待被调换
public void push(int x) {
LinkedList tmp_queue;
if(queue2.isEmpty()){
queue1.offer(x);
}else{
queue2.offer(x);
}
}
//使用队列元素移到等待队列中,最后一个元素不移
public int pop() {
int tmp = 0;
if(queue2.isEmpty()){
while(!queue1.isEmpty()){
tmp = queue1.poll();
if(!queue1.isEmpty()){
queue2.offer(tmp);
}
}
}else{
while(!queue2.isEmpty()){
tmp = queue2.poll();
if(!queue2.isEmpty()){
queue1.offer(tmp);
}
}
}
return tmp;
}
//使用队列元素移到等待队列中,最后一个元素也移动
public int top() {
int tmp = 0;
if(queue2.isEmpty()){
while(!queue1.isEmpty()){
tmp = queue1.poll();
queue2.offer(tmp);
}
}else{
while(!queue2.isEmpty()){
tmp = queue2.poll();
queue1.offer(tmp);
}
}
return tmp;
}
//双空则空
public boolean empty() {
if(queue1.isEmpty() && queue2.isEmpty())return true;
return false;
}
}
另外一个取巧的办法是直接在类中各方法维护一个堆栈。