L2-045 堆宝塔 - 技术栈

原题链接：https://pintia.cn/problem-sets/994805046380707840/exam/problems/1649748772841508872?type=7&page=1

题目描述

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

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

思路分析

很明显的栈。调用C++ STL库中的stack的相关操作即可。

stack<T> s； //定义一个栈，T可以为int，float，double，char，string......

s.push(x); //把x放到栈顶。

s.top(); //返回栈顶的元素

s.pop(); //删除栈顶的元素

s.size(); //返回栈中元素的个数。

s.empty(); //检查栈是否为空，如果为空返回true，否则返回false。

代码实现

cpp 复制代码

#define _CRT_SECURE_NO_WARNINGS 1
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define endl '\n'
const int N = 1010;
stack<int> a, b;
int c[N];

int main() {
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int n; cin >> n;
    for (int i = 1; i <= n; i++) cin >> c[i];
    a.push(c[1]);
    int cnt = 0, h = 0;
    for (int i = 2; i <= n; i++) {
        if (c[i] < a.top()) {
            a.push(c[i]);
        }else {
            if (b.empty() || c[i] > b.top()) {
                b.push(c[i]);
            }else {
                int sa = a.size();
                h = max(h, sa);
                while (!a.empty()) {
                    a.pop();
                }
                cnt++;
                while (!b.empty()&&b.top() > c[i]) {
                    a.push(b.top());
                    b.pop();
                }
                a.push(c[i]);
            }
        }
    }
    int lena = a.size();
    int lenb = b.size();
    h = max({ h,lena,lenb });
    if (a.size()) cnt++;
    if (b.size()) cnt++;
    cout<<cnt<<" "<<h<<endl;
    return 0;
}