当前位置: 代码迷 >> 综合 >> LeetCode 225 Implement Stack using Queues
  详细解决方案

LeetCode 225 Implement Stack using Queues

热度:86   发布时间:2023-10-28 04:49:56.0

思路

只说一下pop和top操作。
(1)对于pop:每次pop的时候:首先将queue中的所有元素依次poll到一个tmp队列中,并用last变量记录最后一个poll出来的元素,用count记录加入tmp队列的次数,用来防止将最后一个需要移除的数据再加进去。最后queue = tmp,返回last即可。
时间复杂度O(n),空间复杂度O(n)
(2)top:与pop操作基本相同,只是不需要用count控制加入tmp队列的次数,因为需要全都加入tmp,最后返回last即可。
时间复杂度O(n),空间复杂度O(n)

代码

class MyStack {
    Queue<Integer> queue;/** Initialize your data structure here. */public MyStack() {
    queue = new LinkedList<>();}/** Push element x onto stack. */public void push(int x) {
    queue.offer(x);}/** Removes the element on top of the stack and returns that element. */public int pop() {
    Queue<Integer> tmp = new LinkedList<>();int last = 0, size = queue.size(), count = 0;while(!queue.isEmpty()) {
    last = queue.poll();if(count < size-1) {
    tmp.offer(last);count++;}}queue = tmp;return last;}/** Get the top element. */public int top() {
    Queue<Integer> tmp = new LinkedList<>();int last = 0;while(!queue.isEmpty()) {
    last = queue.poll();tmp.offer(last);}queue = tmp;return last;}/** Returns whether the stack is empty. */public boolean empty() {
    return queue.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();*/
  相关解决方案