堆宝塔(Python)

news2024/9/26 3:27:39

作者陈越

单位浙江大学

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

首先准备两根柱子，一根 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

代码长度限制

n = int(input())
arr = list(map(int,input().split()))
A = []
B = [] #这个是B柱子，用来存储的
date = [] #用来存储成平塔的喔
while 1:
    # 第一次玩，两根柱子都是空的喔：
    if len(A) == 0 and len(B) == 0:
        A.append(arr[0])
        arr.pop(0)
    elif len(A) != 0:
        if A[-1] > arr[0]:
            A.append(arr[0])
            arr.pop(0)
        elif len(B) == 0:
            B.append(arr[0])
            arr.pop(0)
        elif arr[0] > B[-1]:
            B.append(arr[0])
            arr.pop(0)
        else:
            date.append(len(A))
            A = []
            while len(B) > 0:
                if B[-1] > arr[0]:
                    A.append(B[-1])
                    B.pop(-1)
                else:
                    break
            if len(A) > 0:
                if len(arr) > 0:
                    if A[-1] > arr[0]:
                        A.append(arr[0])
                    else:
                        B.append(arr[0])
                    arr.pop(0)
            else:
                if len(B) == 0:
                    break
                else:
                    A.append(B[-1])
                    B.pop(-1)
                    while len(B) > 0:
                        if B[-1] > arr[0]:
                            A.append(B[-1])
                            B.pop(-1)
                        else:
                            break
    if len(arr) == 0:
        date.append(len(A))
        if len(B) > 0:
            date.append(len(B))
        break
print(len(date),max(date))