设计getMin功能的栈

题目描述：

实现一个特殊功能的栈，在实现栈的基本功能的基础上，再实现返回栈中最小元素的操作。

输入描述：

第一行输入一个整数N，表示对栈进行的操作总数。

下面N行每行输入一个字符串S，表示操作的种类。

如果S为"push"，则后面还有一个整数X表示向栈里压入整数X。

如果S为"pop"，则表示弹出栈顶操作。

如果S为"getMin"，则表示询问当前栈中的最小元素是多少。

输出描述：

对于每个getMin操作，输出一行表示当前栈中的最小元素是多少。

示例1

输入

6
push 3
push 2
push 1
getMin
pop
getMin

输出

1
2

备注：

$1\le N\le 10^6$

$-10^6\le x\le 10^6$

数据保证没有不合法的操作

解法：

法1：

可以使用两个栈进行模拟，一个栈用来保存当前栈中的元素，另一个栈用来保存每一步的最小值，设为 stackData和 stackMin 。设当前数据为val，在入栈时，若val小于等于stackMin栈顶元素，则将val压入stackMin栈中，否则不操作；在出栈时，若stackData栈顶元素等于stackMin栈顶元素，则stakcMin栈顶元素也要出栈；查询最小值则直接取stackMin栈顶元素即可。该方法需要两个栈进行操作，且出栈操作较麻烦。

法2：

使用一个栈进行模拟，用来保存每一步的最小值，设为 stackMin 。设当前数据为val，在入栈时，若val小于等于stackMin栈顶元素，则将val压入stackMin栈中，否则将stackMin栈顶元素重复压入stackMin中；查询最小值则直接取stackMin栈顶元素即可。该方法需要一个栈进行操作，但此法stackMin消耗空间要大于法1中的stackMin。

这里我使用的是法2，并且在stackMin中压入一个无穷大，便面后面判断是否为空的操作。

代码：

#include <cstdio>
#include <algorithm>

using namespace std;

const int N = 1000010;
const int INF = 0x3f3f3f3f;
int stk[N];

int main(void) {
    int n;
    char op[7];
    int val;
    scanf("%d", &n);
    int top = 1;
    stk[0] = INF;
    while (n--) {
        scanf("%s", op);
        if (op[1] == 'u') {
            scanf("%d", &val);
            stk[top] = min(val, stk[top - 1]);
            ++top;
        } else if (op[0] == 'g') printf("%d\n", stk[top - 1]);
        else --top;
    }
    return 0;
}