L2-032 彩虹瓶（25 分)

最新推荐文章于 2026-02-01 09:48:04 发布

原创最新推荐文章于 2026-02-01 09:48:04 发布 · 1.4k 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#天梯赛 #堆栈

天梯赛专栏收录该内容

7 篇文章

订阅专栏

这篇博客讨论了如何解决L2-032彩虹瓶问题，该问题要求按照特定顺序堆放小球。当需要编号为m的小球时，如果送来的正是m，就继续检查m+1；否则，如果栈顶小球编号为m，也可继续；否则，若栈还能放球则放入，否则失败。解决方案通过堆栈实现模拟过程。

题目链接

题目大意：彩虹瓶有n层，必须按照1，2，3…n的顺序堆放小球。假设现在需要编号为m的小球，此时送来的恰好是m，则继续判断m+1层；若不是，如果此时货架上有小球，并且顶层就是m号，则继续判断m+1层。若不是，如果货架上还能放小球，则放小球，否则无法成功制作彩虹瓶。

思路：利用堆栈进行模拟。

提交代码：

#include<bits/stdc++.h>
using namespace std;
int main(){
    int n,m,k;
    cin>>n>>m>>k;
    for(int i=1;i<=k;i++){
        vector<int>v(n+2);
        for(int j=1;j<=n;j++){
            cin>>v[j];
        }
        int ok=1, A=1,B=1;
        stack<int>s;
        while(A<=n){
            if(B<=n&&A==v[B]){ A++;B++;}
            else if(!s.empty()&&s.top()==A){ s.pop(); A++;}
            else if(s.size()<m&&B<=n){ s.push(v[B++]);}
            else { ok=0 ; break; }
        }
        if(ok)cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }

    return 0;
}

带注释代码：

#include<iostream>
#include<vector>
#include<stack>
#include<bits/stdc++.h>

using namespace std;
int main(){
    int n,m,k;
    cin>>n>>m>>k;
    for(int i=1;i<=k;i++){
        vector<int>v(n+2);
        for(int j=1;j<=n;j++){
            cin>>v[j];
        }
        int ok=1, A=1,B=1;
        stack<int>s;
    //注意B的范围，凡是用到v[B]的，都得先判断B的下标
        while(A<=n){
            if(B<=n&&A==v[B]){ A++;B++;}  //送来的正好等于需要的
            else if(!s.empty()&&s.top()==A){ s.pop(); A++;}  //栈顶元素正好是所需要的
            else if(s.size()<m&&B<=n){ s.push(v[B++]);} //栈没满，往栈里面放
            else { ok=0 ; break; }  //无法按照序列输出
        }
        if(ok)cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }

    return 0;
}