天天看點

[每日一道小算法(四)] [棧和隊列] 用兩個棧實作隊列

前言:

哈哈,我有來啦打卡啦,不廢話啦,直接進入正題,哼。

題目描述

用兩個棧來實作一個隊列,完成隊列的Push和Pop操作。 隊列中的元素為int類型。

題目解析

解決這道題之前,我們先來了解兩個概念。

棧:是先進後出的。

隊列:是先進先出的。

明天了這兩個概念就可以做這道題了。

當我們向模拟的隊列插入數 a,b,c 時,假設插入的是 stack1,此時的棧情況為:

棧 stack1:{a,b,c}

棧 stack2:{}

當需要彈出一個數,根據隊列的"先進先出"原則,a 先進入,則 a 應該先彈出。但是此時 a 在 stack1 的最下面,将 stack1 中全部元素逐個彈出壓入 stack2,現在可以正确的從 stack2 中彈出 a,此時的棧情況為:

棧 stack1:{}

棧 stack2:{c,b}

繼續彈出一個數,b 比 c 先進入"隊列",b 彈出,注意此時 b 在 stack2 的棧頂,可直接彈出,此時的棧情況為:

棧 stack1:{}

棧 stack2:{c}

此時向模拟隊列插入一個數 d,還是插入 stack1,此時的棧情況為:

棧 stack1:{d}

棧 stack2:{c}

彈出一個數,c 比 d 先進入,c 彈出,注意此時 c 在 stack2 的棧頂,可直接彈出,此時的棧情況為:

代碼樣例:

import java.util.Stack;
public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
 
    public void push(int node) {
        stack1.push(node);
    }
 
    public int pop() {
        if (stack2.size() <= 0) {
            while (stack1.size() != 0) {
                stack2.push(stack1.pop());
            }
        }
        return stack2.pop();
    }
}      

繼續閱讀