7-1 胡同 (栈练习)

有一个死胡同,宽度刚好只能让一辆汽车通过,偏偏老有汽车开到死胡同来,这下麻烦了,最先开来的汽车要最后才能倒退出去。给定一个汽车开来的序列和一个可能的倒车出去的序列,请判断汽车能否都倒退出去,若能则输出Yes,否则输出No。

输入格式:

首先输入一个整数T,表示测试数据的组数,然后是T组测试数据。每组测试数据首先输入一个正整数n(n≤10),代表开来的汽车数,然后输入2n个整数,其中,前n个整数表示汽车开来的序列,后n个整数表示汽车可能倒出的序列。

输出格式:

对于每组测试,判断能否倒车出该死胡同,若能则输出“Yes”,否则输出“No”。引号不必输出。

输入样例:

2
4 1 2 3 4 2 1 4 3
4 1 2 3 4 4 2 1 3

输出样例:

Yes
No

提示:

样例1的可能方案为:1进,2进,2出,1出,3进,4进,4出,3出。

来源:

[1] 黄龙军,等. 数据结构与算法, 上海:上海交通大学出版社, 2022.7. ISBN: 9787313269881

[2] 黄龙军, 等. 数据结构与算法(Python版),上海: 上海交通大学出版社, 2023. (In Press)

代码长度限制16 KB

时间限制400 ms

内存限制64 MB

代码:

#include<iostream>
#include <memory.h>
#include <stack>
using namespace std;

int main()
{
    int T;
    cin >> T;
    while(T--)
    {
        int n;
        cin >> n;
        int a[n],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;  //定义栈
        //先按顺依次入栈,判断每个进入数与倒出的数是否匹配
        //如果匹配就出栈,不匹配入栈下一个数再进行匹配,
        //最后所有数匹配完了,栈不为空就是不行,为空就是行
        int j = 0;
        for(int i = 0; i < n; i++)
        {
            //cout << "i:" << i << endl;
            s.push(a[i]);
            //如果栈顶元素和倒车顺序匹配就出栈,并且开始比对下一个数
            while(!s.empty() && b[j] == s.top())
            {
                s.pop();
                //cout << "j:" << j << endl;
                j++;

            }
        }
        if(s.empty())
        {
            cout << "Yes"<< endl ;
        }
        else
        {
            cout << "No" << endl;
        }

    }

    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_63484669/article/details/128666866
7-1
今日推荐