LeetCode Implement Queue using Stacks

来源:互联网 发布:软件需求分析报告实例 编辑:程序博客网 时间:2024/05/01 14:06

原题链接在这里:https://leetcode.com/problems/implement-queue-using-stacks/

本题与Implement Stack using Queues相对应。

用Stack implement queue时,可以采用两个stack,add时就是一直像stk1中压栈。

poll()时把stk1的所有元素逐个压入另一个stack stk2中,现在stk2最上面的元素就是最先来的的,pop()之后再把剩下的压回到stk1中。

peek()与poll()类似,但这次不用pop(), 只需peek()出一个值返回。

Note: 1. 中间peek()时 用到了int res = stk2.peek(); 

所以生成stack时一定要加上类型<Integer>, if using Stack stk2 = new Stack(), it will throw error.

AC Java:

class MyQueue {    // Push element x to the back of queue.    private Stack<Integer> stk1 = new Stack<Integer>();    private Stack<Integer> stk2 = new Stack<Integer>();    public void push(int x) {        stk1.push(x);    }    // Removes the element from in front of queue.    public void pop() {        while(!stk1.empty()){            stk2.push(stk1.pop());        }        stk2.pop();        while(!stk2.empty()){            stk1.push(stk2.pop());        }    }    // Get the front element.    public int peek() {        while(!stk1.empty()){            stk2.push(stk1.pop());        }        int res = stk2.peek();        while(!stk2.empty()){            stk1.push(stk2.pop());        }        return res;    }    // Return whether the queue is empty.    public boolean empty() {        return stk1.empty();    }}


0 0
原创粉丝点击