题解 | 验证栈序列

#include <bits/stdc++.h>

using namespace std;

int main() {
    int q;
    cin>>q;
    while(q--)
    {
        int n;
        cin>>n;
        vector<int>a(n);
        vector<int>b(n);
        for(int i=0;i<n;i++)cin>>a[i];
        for(int i=0;i<n;i++)cin>>b[i];

        stack<int>s;
        bool is_ok=true;
        int j=0;
        int cnt=0;
        for(int i=0;i<n;i++)
        {
            s.push(a[i]);
            while(!s.empty()&&s.top()==b[j])
            {
                s.pop();
                j++;
                cnt++;//pop了几个数
            }
        }

        if(cnt<n)
        {
            for(;j<n;j++)
            {
                if(b[j]==s.top())
                {
                    cnt++;
                    s.pop();
                }
                else 
                {
                    is_ok=false;
                    break;
                }
            }   
        }

        if(is_ok&&cnt==n)
        cout<<"Yes"<<endl;
        else
        cout<<"No"<<endl;
    }
    return 0;
}
// 64 位输出请用 printf("%lld")