• 232.Implement Queue using Stacks



    用栈实现队列的功能。
    MyQueue queue = new MyQueue();

    queue.push(1);
    queue.push(2);
    queue.peek(); // returns 1
    queue.pop(); // returns 1
    queue.empty(); // returns false

    思路:
    运用2个栈s_tmp,s ,因为栈是后进先出,队列是先进先出,所以,将每一次push的元素 x 都放到栈的栈底,先将栈 s 中的元素依次放入 s_tmp 中,再将 x 放入 s_tmp中,最后将 s_tmp中的元素再放入 s 中,则, 栈 s 中的元素始终都是最先进的在栈顶,最后进的在栈底。

    class MyQueue {
    private:
        stack<int> s_tmp, s;
    public:
        void push(int x) {
            while (!s.empty()) {
                s_tmp.push(s.top());
                s.pop();
            }
            s_tmp.push(x);
            while (!s_tmp.empty()) {
                s.push(s_tmp.top());
                s_tmp.pop();
            }
        }
        int pop() {
            int res = s.top();
            s.pop();
            return res;
        }
        int peek() {
            return s.top();
        }
        bool empty() {
            return s.empty();
        }
    };

    Java 版:

      • 利用两个栈来做,插入的时候,都插入到 stack2 中;
      • stack2 中,保持存一个元素的大小,超过1个,就放入 stack1 中;
      • 当取队列元素的时候,利用 stack2 中转,拿到 stack1 最底层的元素;
      • 再将 stack2 中转后的元素,放入 stack1 中。

    class MyQueue {
        
        /** Initialize your data structure here. */
        Stack<Integer> stack1;
        Stack<Integer> stack2;
        public MyQueue() {
            stack1 = new Stack<>();
            stack2 = new Stack<>();
        }
        
        /** Push element x to the back of queue. */
        public void push(int x) {
            if(stack2.isEmpty()) stack2.push(x);
            else{
                stack1.push(stack2.pop());
                stack2.push(x);
            }
        }
        
        /** Removes the element from in front of queue and returns that element. */
        public int pop() {
            int res;
            while(!stack1.isEmpty()){
                stack2.push(stack1.pop());
            }
            res = stack2.pop();
            while(!stack2.isEmpty()){
                stack1.push(stack2.pop());
            }
            return res;
        }
        
        /** Get the front element. */
        public int peek() {
            int res;
            while(!stack1.isEmpty()){
                stack2.push(stack1.pop());
            }
            res = stack2.peek();
            while(!stack2.isEmpty()){
                stack1.push(stack2.pop());
            } 
            return res;
        }
        
        /** Returns whether the queue is empty. */
        public boolean empty() {
            return stack1.isEmpty() && stack2.isEmpty();
        }
    }
  • 相关阅读:
    Java知识系统回顾整理01基础04操作符02关系操作符
    Java知识系统回顾整理01基础04操作符01算术操作符
    Java知识系统回顾整理01基础03变量09块
    Java知识系统回顾整理01基础03变量08表达式
    Java知识系统回顾整理01基础03变量07final关键字
    Java知识系统回顾整理01基础03变量06变量的作用域
    Java知识系统回顾整理01基础03变量05变量命名规则
    Java知识系统回顾整理01基础03变量04类型转换
    leetcode-----64. 最小路径和
    leetcode-----63. 不同路径 II
  • 原文地址:https://www.cnblogs.com/luo-c/p/12876582.html
Copyright © 2020-2023  润新知