2019年天梯赛全国总决赛题集 L2-4 彩虹瓶（25 分)

最新推荐文章于 2022-03-23 20:16:44 发布

1直守护你

最新推荐文章于 2022-03-23 20:16:44 发布

阅读量958

点赞数 2

分类专栏：算法文章标签： d

本文链接：https://blog.csdn.net/qq_41512889/article/details/88937650

版权

算法专栏收录该内容

3 篇文章 0 订阅

订阅专栏

本文介绍了一个名为L2-4彩虹瓶的编程挑战，该挑战要求使用两个栈来模拟瓶子中颜色的排序过程，并判断是否能在一定条件下将所有颜色按顺序放入一个栈中。此程序通过读取输入的颜色序列并运用特定的算法进行处理，最终输出是否成功完成颜色排序。

摘要由CSDN通过智能技术生成

L2-4 彩虹瓶（25 分)

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

int main()
{
    int N,M,K;
    cin>>N>>M>>K;
    int p[1000];
    int i;
    for(i=0;i<K;i++)
    {
        stack<int> s1,s2;
        bool over=false;
        for(int j=0;j<N;j++)
            cin>>p[j];
        s1.push(0);
        for(int j=0;j<N;j++)
        {
            if(!s2.empty()&&s2.top()==s1.top()+1&&s2.size()<=M)
            {
                s1.push(s2.top());
                s2.pop();
                j--;
                continue;
            }
            if(p[j]==s1.top()+1)
            {
                s1.push(p[j]);
            }
            else
            {
                s2.push(p[j]);
            }
            if(s2.size()>M)
            {
                over=true;
                cout<<"NO"<<endl;
                break;
            }
        }
        if(!over)
        while(!s2.empty())
        {
            if(s2.top()==s1.top()+1)
            {
                s1.push(s2.top());
                s2.pop();
            }
            else
            {
                cout<<"NO"<<endl;
                break;
            }
        }
        if(s2.empty())
            cout<<"YES"<<endl;
    }
    return 0;
}