51Nod 2160 数字游戏 刷题笔记

1. 题目描述

1.1. Limit

Time Limit: 1000 ms

Memory Limit: 131072 kB

1.2. Problem Description

现在有$n$个数字依次进入一个栈，每个数字$a$进入栈的时候，如果栈顶元素小于$a$，则会将栈顶元素弹出，新的栈顶元素如果仍然小于$a$，则会将新的栈顶元素继续弹出，直到栈顶元素大于等于$a$为止，$a$才会加入栈。问$n$个数字依次进入后，最后栈里的数字依次是？

---

1.3. Input

输入第一行一个整数$n(n\le 100000)$，表示有$n$个数字依次进入栈。
接下来$n$行，每行一个整数，表示第$i$个数字。

---

1.4. Output

若干行，表示最后栈中的数字。

---

1.5. Sample Input

5
5
3
2 
4
1

1.6. Sample Output

5
4
1

1.7. Source

51Nod 2160 数字游戏

---

2. 解读

使用STL中的 Vector 按照题目中的要求模拟栈的入栈出栈即可。

注：在判断栈顶元素是否小于即将入栈的元素的for循环中，不能使用 i < stack.size() 来判断是否跳出循环，因为出栈时 stack.size() 会减1，这样使用会导致循环最多只能遍历一半栈中的元素。

3. 代码

#include <iostream>
#include <string.h>
#include <vector>
using namespace std;

int main()
{
    long long n;
    // 读入n
    scanf("%lld", &n);
    // 初始化栈
    vector<int> stack;
    // 栈的深度
    long long stackDepth = 0l;
    // buffer
    long long buffer;
    // 若大于0
    if (n > 0) {
        // 读取第一个元素
        scanf("%lld", &buffer);
        // 入栈
        stack.push_back(buffer);
        // 深度加1
        stackDepth++;
        // 读入入栈元素
        for (long long i = 1; i < n; i++) {
            // 读取输入
            scanf("%lld", &buffer);
            size_t sizeBuffer = stackDepth;
            for (size_t j = 0; j < sizeBuffer; j++) {
                // 若栈顶元素小于即将新入栈的元素buffer
                if (stack[stackDepth - 1] < buffer && stackDepth > 0) {
                    // 栈顶元素出栈
                    stack.pop_back();
                    // 深度减1
                    stackDepth--;
                } else {
                    break;
                }
            }
            // 元素入栈
            stack.push_back(buffer);
            // 深度加1
            stackDepth++;
        }
        // 输出栈内元素
        for (size_t i = 0; i < stack.size(); i++) {
            printf("%d\n", stack[i]);
        }
    }
}

---

联系邮箱：curren_wong@163.com

Github：https://github.com/CurrenWong

欢迎转载/Star/Fork，有问题欢迎通过邮箱交流。