PTA L2-045 堆宝塔

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

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

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

输入格式：

输入第一行给出一个正整数 N（≤103），为彩虹圈的个数。第二行按照宝宝抓取的顺序给出 N 个不超过 100 的正整数，对应每个彩虹圈的直径。

输出格式：

在一行中输出宝宝堆出的宝塔个数，和最高的宝塔的层数。数字间以 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

做法：

1.拿一个彩虹圈t

        1.1.A可放，放A上

        1.2.B可放，放B上

        1.3.B不可放，完成一件宝塔制作，记录数量和最高高度,把B所有的大于t的放到A上，

         并把t放到A上，整理B。

代码：

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int N = 1010;

int B[N],A[N],id_a,id_b;

int main()
{
    int n = 0,cnt = 0,h = 0;
    scanf("%d",&n);
    for(int i = 0;i < n;i++)
    {
        int t = 0;
        scanf("%d",&t);
        if(!id_a || t < A[id_a - 1]) A[id_a++] = t;//A可放
        else
        {
            if(!id_b || B[id_b - 1] < t) B[id_b++] = t;//B可放
            else//完成一件宝塔制作
            {
                cnt++;
                h = max(h,id_a);//记录高度
                id_a = 0;
                for(int i = id_b - 1;i >= 0;i--)//B所有的大于t的放到A上
                    if(B[i] > t) A[id_a++] = B[i],B[i] = 0;
                for(int i = 0;i < id_b;i++)//整理B的内部空间，把不为0的数往前放
                {
                    if(B[i] == 0)
                    {
                        int j = i + 1;
                        while(j < id_b && !B[j])j++;
                        if(j < id_b) B[i] = B[j],B[j] = 0;
                        else
                        {
                            id_b = i;
                            break;
                        }
                    }
                }
                A[id_a++] = t;//放t
            }
        }
    }
  
    if(id_a) cnt++,h = max(h,id_a);//处理剩余的
    if(id_b) cnt++,h = max(h,id_b);
    printf("%d %d",cnt,h);
    return 0;
}

结果：