PTA | 堆宝塔

目录

题目：

输入格式：

输出格式：

输入样例：

输出样例：

样例解释：

代码： 

无注释版：

有注释版：

---

题目：

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

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

代码长度限制16 KB，Java (javac)时间限制500 ms，内存限制256 MB，其他编译器时间限制400 ms，内存限制64 MB，栈限制8192 KB

代码： 

C++（clang++)

无注释版：

#include<bits/stdc++.h>
using namespace std;
#define int long long
stack<int> a,b;
signed main(){
	int n;
	cin>>n;
	int ma=0;
	int cnt=0;
	while(n--){
		int x;
		cin>>x;
		if(a.empty()){
			a.push(x);
			continue;
		}
		int t=a.top();
		if(t>x){
			a.push(x);
			continue;
		}	
		if(b.empty()){
			b.push(x);
			continue;
		}
		int tt=b.top();
		if(tt<x){
			b.push(x);
			continue; 
		}
		int s=0;
		while(!a.empty()){
			a.pop();
			s++;
		}
		ma=max(ma,s);
		cnt++;
		while(!b.empty()&&x<b.top()){
			a.push(b.top());
			b.pop();
		} 
		a.push(x);
	}
	int s=0;
	if(!b.empty()){
		cnt++;
	}
	if(!a.empty()){
		cnt++;
	}
	while(!b.empty()){
		b.pop();
		s++;
	}
	ma=max(ma,s);
	s=0;
	while(!a.empty()){
		a.pop();
		s++;
	}
	ma=max(ma,s);
	cout<<cnt<<" "<<ma<<"\n";
}

有注释版：

#include<bits/stdc++.h>
using namespace std;
#define int long long  // 使用 long long 来处理大整数
stack<int> a,b;  // 使用两个栈 a 和 b，a 用于主宝塔，b 用于临时存放
signed main(){
    int n;
    cin >> n;  // 输入彩虹圈的数量
    int ma = 0;  // 记录最高的宝塔层数
    int cnt = 0;  // 记录宝塔的数量

    while (n--) {  // 每次处理一个彩虹圈
        int x;
        cin >> x;  // 输入当前彩虹圈的直径

        if (a.empty()) {  // 如果 A 栈为空，直接放入 A 栈
            a.push(x);
            continue;  // 进入下一个彩虹圈
        }

        int t = a.top();  // 获取 A 栈顶部的彩虹圈直径
        if (t > x) {  // 如果当前彩虹圈比 A 栈顶部小
            a.push(x);  // 直接放入 A 栈
            continue;  // 进入下一个彩虹圈
        }

        if (b.empty()) {  // 如果 B 栈为空，直接放入 B 栈
            b.push(x);
            continue;  // 进入下一个彩虹圈
        }

        int tt = b.top();  // 获取 B 栈顶部的彩虹圈直径
        if (tt < x) {  // 如果当前彩虹圈比 B 栈顶部大
            b.push(x);  // 放入 B 栈
            continue;  // 进入下一个彩虹圈
        }

        int s = 0;  // 用于统计当前从 A 栈中取出的彩虹圈数量，即当前堆积的宝塔层数
        while (!a.empty()) {  // 将 A 栈中的所有彩虹圈取出
            a.pop();
            s++;  // 每取出一个彩虹圈，层数增加
        }

        ma = max(ma, s);  // 更新最高宝塔的层数
        cnt++;  // 宝塔数增加

        // 将 B 栈中的所有比当前彩虹圈小的彩虹圈放回 A 栈
        while (!b.empty() && x < b.top()) {  
            a.push(b.top());
            b.pop();  // 取出 B 栈顶部的彩虹圈，并放入 A 栈
        }

        a.push(x);  // 将当前彩虹圈放入 A 栈
    }

    int s = 0;
    if (!b.empty()) {  // 如果 B 栈中还有彩虹圈，说明还有一个宝塔
        cnt++;  // 宝塔数增加
    }

    if (!a.empty()) {  // 如果 A 栈中还有彩虹圈，说明还有一个宝塔
        cnt++;  // 宝塔数增加
    }

    while (!b.empty()) {  // 统计 B 栈中剩余的彩虹圈数量，即堆积成的宝塔层数
        b.pop();
        s++;  // 每取出一个彩虹圈，层数增加
    }
    ma = max(ma, s);  // 更新最高宝塔的层数

    s = 0;  
    while (!a.empty()) {  // 统计 A 栈中剩余的彩虹圈数量，即堆积成的宝塔层数
        a.pop();
        s++;  // 每取出一个彩虹圈，层数增加
    }
    ma = max(ma, s);  // 更新最高宝塔的层数

    // 输出最终的宝塔数量和最高宝塔层数
    cout << cnt << " " << ma << "\n";
}