2023程序设计天梯赛 L2-1 堆宝塔

第一篇题解就拿这题开始吧

题目

L2-1 堆宝塔 分数 25

作者 陈越

单位 浙江大学

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

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

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

输入格式：

输入第一行给出一个正整数 （ ），为彩虹圈的个数。第二行按照宝宝抓取的顺序给出  个不超过 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

题目分析

题目给我们这些信息

• 彩虹圈的个数与直径，固定的A，B两根柱子
• 彩虹圈的摆放规则
  • 1. 比A柱最上面的小(或A柱空)则直接放上A柱
  • 2. 若不满足1，与B柱最上面的比，若大(或B柱空)，则放B柱
  • 3. 不满足1、2，则取下A柱上所有彩虹圈作为一件成品，B柱反转逐一放入A柱，放入时只放比新彩虹圈大的
  • 4. 重复以上步骤直到彩虹圈耗尽

输入与输出

我们需要处理的输入就两种，首先是彩虹圈总个数  ；然后是各个彩虹圈的直径。

输出两个数，第一个数是成品个数；第二个数是最高层数。

解题步骤

解决这题的方法很明朗，其实就是将每个彩虹圈按照规则对比摆放即可。

那A、B两柱如何实现呢？

观察柱子的行为，柱子主要的作用之一是存彩虹圈，彩虹圈放入则需要用到柱子最顶端的彩虹圈，这不恰好用上了栈的特性先进后出吗？

那么，最简单的思路呼之欲出：

首先我们用两个stack来作为A、B两柱：

stack<int> a, b;

按照彩虹圈摆放规则写出核心代码：

while (n--) {
	int x;
	cin >> x; // 输入每个彩虹圈的直径
	if (a.empty()) a.push(x); // 如果a柱子是空的，就在a柱子上放好
	else if (x < a.top()) a.push(x); // 如果x比a柱子最上面的彩虹圈小，就直接放上去
	else if (b.empty() || x > b.top()) b.push(x); // 如果b柱子是空的或者x大，就在b柱子上放好
	else
	{
		towerCount++; // 把a柱子上串好的宝塔取下来作为一件成品，增加宝塔的数量
		mx = max(mx, (int)a.size()); // 更新最高的宝塔的层数
		while(!a.empty()) a.pop(); // 清空a柱子
		while(!b.empty() && b.top() > x) // 把b柱子上所有比x大的彩虹圈逐一取下放到a柱子上
		{
			a.push(b.top());
			b.pop();
		}
		a.push(x); // 最后把x也放到a柱子上
	}
}

最后一个彩虹圈单独讨论：

if(!a.empty()) {
	towerCount++; // 最后a柱子上剩下的宝塔作为一件成品，增加宝塔的数量
	mx = max(mx, (int)a.size()); // 更新最高的宝塔的层数
}

if(!b.empty()) {
	towerCount++; // b柱子上剩下的彩虹圈被逐一取下，堆成另一座宝塔，增加宝塔的数量
	mx = max(mx, (int)b.size()); // 更新最高的宝塔的层数
}

完整代码

#include<bits/stdc++.h>
using namespace std;

int main()
{
	stack<int> a,b; // a和b分别代表两根柱
	int towerCount = 0; // 宝塔的数量
	int mx = 0; // 最高的宝塔的层数

	int n;
	cin >> n;
	while (n--)
	{
		int x;
		cin >> x;
		if (a.empty()) a.push(x); // 如果a柱空，就放a柱
		else if (x < a.top()) a.push(x); // 如果x小，直接放
		else if (b.empty() || x > b.top()) b.push(x); // 如果b柱空或x大，就放b柱
		else
		{
			towerCount++; // 将a柱作为一件成品
			mx = max(mx, (int)a.size());
			while(!a.empty()) a.pop(); // 清空a柱
			while(!b.empty() && b.top() > x) // 把b柱上所有比x大的彩虹圈逐一取下放到a柱子上
			{
				a.push(b.top());
				b.pop();
			}
			a.push(x); // 最后把x也放到a柱上
		}
	}

	if(!a.empty()) 
	{
		towerCount++; // 最后a柱上剩下的宝塔作为一件成品，增加宝塔的数量
		mx = max(mx, (int)a.size());
	}
	if(!b.empty()) 
	{
		towerCount++; // 最后b柱上剩下的宝塔作为一件成品，增加宝塔的数量
		mx = max(mx, (int)b.size());
	}

	cout << towerCount << " " << mx << endl;

	return 0;
}