《劍指offer》面試題22——棧的壓入、彈出序列(C++)

棧的壓入、彈出序列(C++)

1、題目描述

輸入兩個整數序列，第一個序列表示棧的壓入順序，請判斷第二個序列是否爲該棧的彈出順序。假設壓入棧的所有數字均不相等。例如序列1,2,3,4,5是某棧的壓入順序，序列4，5,3,2,1是該壓棧序列對應的一個彈出序列，但4,3,5,1,2就不可能是該壓棧序列的彈出序列。（注意：這兩個序列的長度是相等的）

鏈接：https://www.nowcoder.com/questionTerminal/d77d11405cc7470d82554cb392585106
來源：牛客網

2、思路

借用一個輔助的棧，遍歷壓棧順序，先講第一個放入棧中，這裏是1，然後判斷棧頂元素是不是出棧順序的第一個元素，這裏是4，很顯然1≠4，所以我們繼續壓棧，直到相等以後開始出棧，出棧一個元素，則將出棧順序向後移動一位，直到不相等，這樣循環等壓棧順序遍歷完成，如果輔助棧還不爲空，說明彈出序列不是該棧的彈出順序。
舉例：
入棧1,2,3,4,5
出棧4,5,3,2,1
首先1入輔助棧，此時棧頂1≠4，繼續入棧2
此時棧頂2≠4，繼續入棧3
此時棧頂3≠4，繼續入棧4
此時棧頂4＝4，出棧4，彈出序列向後一位，此時爲5，,輔助棧裏面是1,2,3
此時棧頂3≠5，繼續入棧5
此時棧頂5=5，出棧5,彈出序列向後一位，此時爲3，,輔助棧裏面是1,2,3。
….
依次執行，最後輔助棧爲空。如果不爲空說明彈出序列不是該棧的彈出順序。

3、代碼

#include<iostream>
#include <vector>
using namespace std;

bool IsPopOrder(vector<int> pushV,vector<int> popV)
{
    if(pushV.size()!=popV.size())  return false;

    vector<int> vecstack;
    for(int i = 0, j = 0; i < pushV.size(); i++)
        {
             vecstack.push_back(pushV[i]);                                         //vec.push_back   在尾部加入一個數據
             while( j < popV.size() && vecstack.back() == popV[j])   //vc.back()    傳回最後一個數據
                {
                    vecstack.pop_back();                                                //vec.pop_back()          //刪除最後一個數據。
                    j++;
                }
        }
    return vecstack.empty();
}

void Print(vector<int> &myarray)
{
    int num=myarray.size();
    for(int i=0; i<num;i++)
         cout<<myarray[i]<<"  ";
    cout<<endl;
    return;
}

vector<int> Arr2Vec(int myarray[],int num)
{
    vector<int> vec;
    vec.clear();
    for(int i=0;i<num;i++)
        vec.push_back(myarray[i]);

    return vec;
}
int main()
{
    int array1[]={1,2,3,4,5};
    int array2[]={4,5,3,2,1};
    int num1=sizeof(array1)/sizeof(int);
    int num2=sizeof(array2)/sizeof(int);

    vector<int> pushV=Arr2Vec(array1,num1);
    vector<int> popV= Arr2Vec(array2,num2);

    cout<<"pushV:"<<endl;
    Print(pushV);
    cout<<"popV:"<<endl;
    Print(popV);
    cout<<endl;
    cout<<IsPopOrder(pushV,popV)<<endl;
    return 0;
}

運行結果爲：

pushV:
1  2  3  4  5
popV:
4  5  3  2  1