设计一个有getMin功能的栈

使用C++完成《程序员代码面试指南 IT名企算法与数据结构题目》

题目

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

实现

使用两个栈，一个栈用来保存当前栈中的元素，起功能和一个正常的栈没有区别，这个栈记为stackData；另一个栈用于保存每一步的最小值，这个栈记为stackMin

1. 第一种方案

压入数据规则：

假设当前数据为newNum，先将其压入stackData，然后判断stackMin是否为空：
- 如果为空，则newNum压入stackMin
- 如果不为空，则比较newNum和stackMin的栈顶元素中哪一个更小
- 如果newNum更小或者两者相等，则newNum也压入stackMin
- 如果stackMin中栈顶元素小，则stackMin不压入任何内容

弹出数据规则

• 先在stackData中弹出栈顶元素，记为value。然后比较当前stackMin的栈顶元素和value哪一个更小
• 当value等于stackMin的栈顶元素时，stackMin弹出栈顶元素
• 当value大于stackMin的栈顶元素时，stackMin不弹出栈顶元素：返回value

getMin操作（查询当前栈中的最小值操作）

由于stackMin始终记录着stackData中的最小值，所以，stackMin的栈顶元素始终是当前stackData中的最小值

方案1实现代码

class MyStack1{
private:
    stack <int>stackData;
    stack <int>stackMin;
public:
    MyStack1(){}
    void push(int newNum){
        if(stackMin.empty()){
            stackMin.push(newNum);
        }else if (newNum <= getmin()){
            stackMin.push(newNum);
        }
        stackData.push(newNum);
    }
    int pop(){
        if(stackData.empty()){
            fprintf(stderr, "The stcack is empty\n");
            exit(0);
        }
        int value = stackData.top();
        stackData.pop();
        if(value == getmin()){
            stackMin.pop();
        }
        return value;
    }
    int getmin(){
        if(stackMin.empty()){
            fprintf(stderr,"Your Stack is empty");
            exit(0);
        }
        return stackMin.top();
    }
};

2. 第二种方案

压入数据规则

• 假设当前数据为newNum，先将其压入stackData，然后判断stackMin是否为空：
• 如果为空，则newNum也压入stackMin
• 如果不为空，则比较newNum和stackMin的栈顶元素中哪一个更小：
• 如果newNum更小或两者相等，则newNum也压入stackMin
• 如果stackMin中栈顶元素小，则把stackMin的栈顶元素重复压入stackMin

弹出数据规则

• 在stackData中弹出数据，弹出的数据记为value；
• 弹出stackMin中的栈顶；
• 返回value

getMin操作（查询当前栈中最小值操作）

由于stackMin始终记录着stackData中的最小值，所以，stackMin的栈顶元素即为当前stackData中的最小值

方案2实现代码

class MyStack2{
private:
    stack <int>stackData;
    stack <int>stackMin;
public:
    MyStack2(){}
    void push(int newNum){
        if(stackMin.empty()){
            stackMin.push(newNum);
        }else if (newNum<getmin()){
            stackMin.push(newNum);
        }else{
            int newMin = stackMin.top();
            stackMin.push(newMin);
        }
        stackData.push(newNum);
    }
    int pop(){
        if(stackData.empty()){
            fprintf(stderr,"The stack is empty");
            exit(0);
        }
        stackMin.pop();
        int popNum=stackData.top();
        stackData.pop();
        return popNum;
    }
    int getmin(){
        if(stackMin.empty()){
            fprintf(stderr, "Your stack is empty");
            exit(0);
        }
        return stackMin.top();
    }
};

第一种方案在stackMin的栈顶只存放当前stackData里的最小值一次，而第二种方案每次在压入和弹出stackData栈时都要压入和弹出stackMin中数据，向stackMin栈中重复压入当前stackData里的最小值