23-05-09--栈和队列--堆宝塔

daniel350-wang / 2023-05-09 / 原文

堆宝塔游戏是让小朋友根据抓到的彩虹圈的直径大小，按照从大到小的顺序堆起宝塔。但彩虹圈不一定是按照直径的大小顺序抓到的。聪明宝宝采取的策略如下：

首先准备两根柱子，一根 A 柱串宝塔，一根 B 柱用于临时叠放。
把第 1 块彩虹圈作为第 1 座宝塔的基座，在 A 柱放好。
将抓到的下一块彩虹圈 C 跟当前 A 柱宝塔最上面的彩虹圈比一下，如果比最上面的小，就直接放上去；否则把 C 跟 B 柱最上面的彩虹圈比一下：
- 如果 B 柱是空的、或者 C 大，就在 B 柱上放好；
- 否则把 A 柱上串好的宝塔取下来作为一件成品；然后把 B 柱上所有比 C 大的彩虹圈逐一取下放到 A 柱上，最后把 C 也放到 A 柱上。

重复此步骤，直到所有的彩虹圈都被抓完。最后 A 柱上剩下的宝塔作为一件成品，B 柱上剩下的彩虹圈被逐一取下，堆成另一座宝塔。问：宝宝一共堆出了几个宝塔？最高的宝塔有多少层？

输入格式：

输入第一行给出一个正整数

输出格式：

在一行中输出宝宝堆出的宝塔个数，和最高的宝塔的层数。数字间以 1 个空格分隔，行首尾不得有多余空格。

输入样例：

11
10 8 9 5 12 11 4 3 1 9 15

输出样例：

4 5

样例解释：

宝宝堆成的宝塔顺次为：

10、8、5
12、11、4、3、1
9
15、9

代码如下：

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

int main()
{
    stack <int> a,b;
    int n;
    cin>>n;
    queue <int> q;
    for(int i=0;i<n;i++)
    {
        int t;
        cin>>t;
        q.push(t); 
    }
    int cnt=0;
    int maxsize=0;
    while(!q.empty() )
    {
        if(a.empty() )
        {
            a.push(q.front() );
            q.pop() ;
        }else{
            if(q.front() < a.top())
            {
                a.push(q.front() );
                q.pop() ;
            }else{
                if(b.empty()||q.front() >b.top() )
                {
                    b.push(q.front());
                    q.pop() ;
                    
                }else{
                    if(a.size() >maxsize)
                    {
                        maxsize=a.size() ;
                    }
                    while(!a.empty() )
                    {
                        a.pop() ;
                    }
                    while(!b.empty() &&b.top() >q.front() )
                    {
                        a.push(b.top() );
                        b.pop() ;
                    }
                    a.push(q.front() );
                    q.pop() ;
                    cnt++;
                }
            }
            
        }
    }
    if(!a.empty())
    {
        if(a.size() >maxsize)
        {
            maxsize=a.size() ;
        }
        cnt++;
    }
    if(!b.empty() )
    {
        if(b.size() >maxsize)
        {
            maxsize=a.size() ;
        }
        cnt++;
    }
    cout<<cnt<<' '<<maxsize;
}