【數據結構】判斷出棧順序合法性

時間 2019-11-12

原文原文鏈接

#include      <iostream>
#include       <iomanip>
#include        <string>
#include        <cstdio>
#include         <cmath>
#include       <cstring>
#include     <algorithm>
#include        <vector>
#include         <queue>
#include         <deque>
#include           <map>
#include <unordered_map>
#include           <set>
#include        <bitset>
#include       <fstream>
#include        <time.h>
#include         <stack>

#define               P(n, f) cout<<n<<(f?'\n':' ')
#define                            pi pair<int,int>
#define                                LL long long
#define                      ULL unsigned long long
#define                            lowbit(x) x&(-x)
#define                                mp make_pair
#define                                elif else if
#define       range(i, a, b) for(auto i=a;i<=b;++i)
#define     itrange(i, a, b) for(auto i=a;i!=b;++i)
#define     rerange(i, a, b) for(auto i=a;i>=b;--i)
#define  fill(arr, tmp) memset(arr,tmp,sizeof(arr))
#define IOS ios::sync_with_stdio(false), cin.tie(0)
using namespace std;
template <class T>
class JudgeGirl{
public:
    bool main(vector<T>a,vector<T>b){
        if(a.empty() or b.empty()){
            P("NO",1);
            return false;
        }
        if(a.size()!=b.size()){
            P("NO",1);
            return false;
        }
        stack<T>S;
        int posA=0,posB=0;
        while(posA<a.size()){
            if(S.empty() or S.top()!=b[posB])S.push(a[posA++]);
            else S.pop(),posB++;
        }
        while(!S.empty()){
            if(S.top()!=b[posB++]){
                P("NO",1);
                return false;
            }
            S.pop();
        }
        P("YES",1);
        return true;
    }
};
JudgeGirl<int> A; //能夠本身改順序表的類型
vector<int>a,b; //同上
int n;

int main() {
    cout<<"順序表長度：";cin>>n;
    range(i,1,n){
        int tmp;
        cin>>tmp;
        a.push_back(tmp);
    }
    range(i,1,n){
        int tmp;
        cin>>tmp;
        b.push_back(tmp);
    }
    A.main(a,b);
    return 0;
}