1. 程式人生 > >【劍指offer - C++/Java】5、用兩個棧實現佇列

【劍指offer - C++/Java】5、用兩個棧實現佇列

牛客網題目連結:用兩個棧實現佇列

題目描述:

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

1、題目分析

首先我們知道資料結構中棧與佇列的特性是:

  • 棧:資料先進後出
  • 佇列:資料先進先出

假設現在佇列的兩個基本操作為push與pop,兩個輔助的棧為stack1與stack2。那麼可以這實現佇列的性質:

當我們push資料的時候,只把資料往stack1中壓入。因為壓入到stack1中的資料是先進後出,順序是反的,所以當需要pop資料時,再將stack1中的資料全部倒入(彈出)到stack2,這樣的話,資料的順序就變回來了,此時再在stack2中彈出資料返回即可。例如往stack1中壓入1-5,此時在stack1中由上往下是5-1,當想pop時,先將 stack1中的5-1倒入stack2,那麼在stack2中由上往下就是1-5,此時再讓stack2彈出1即可,這就模擬了先進先出規則。如下圖所示:

在這裡插入圖片描述

以上只是很簡單的情況(第一次倒資料)。還有這種情況,1-5已經倒入到stack2中,然後stcak1中沒有資料了,如果現在push,就將資料壓入到stack1中。如果再pop呢?首先需要判斷stack2中還有沒有資料,如果stack2中還有資料,則繼續彈出stack2中的資料。如果stack2中沒有資料了,則將stack1中的所有資料全部倒入stack2中,然後從stack2中彈出資料(自己想一下這個過程)。那麼這就是一個完全模擬佇列的過程。

2、程式碼

Java程式碼:

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); //stack1 用於push資料 } /* stack2用於pop資料 */ public int pop() { /* 對於題來說,可以不用判斷異常情況也可以通過 */ if(stack1.isEmpty(
) && stack2.isEmpty()){ throw new RuntimeException("Queue is empty!"); } /* 如果Stack2為空,則將stack1中的資料全部倒進stack2 */ if(stack2.isEmpty()){ while(!stack1.isEmpty()){ stack2.push(stack1.pop()); } } return stack2.pop(); } }

C++程式碼

class Solution
{
public:
    void push(int node) {
        /* stack1作為插入資料的棧 */
        stack1.push(node);
    }
    int pop() {
        /* stack2作為pop資料的棧 */
        int tmp=0;
        if(stack2.empty()){
            while(!stack1.empty()){
                tmp=stack1.top();
                stack1.pop();
                stack2.push(tmp);
            }
        }
        tmp=stack2.top();
        stack2.pop();
        return tmp;
    }
private:
    stack<int> stack1;
    stack<int> stack2;
};

3、總結

理解棧與佇列的push與pop操作。

學習探討加:
qq:1126137994
微信:liu1126137994