程序员面试100题--02

题目描述：

实现一个min操作位O(1)的栈

//之前其实写过一次，这次换一个何海涛的方法，设定辅助栈保存当前的最小值的索引。 
1 #include <iostream>
#include <queue>
#include <climits>
#include <algorithm>
#include <memory.h>
#include <stdio.h>
#include <map>
#include <vector>
#include <list>
#include <stack>
using namespace std;

/*
思路:
每次push的时候，同时push一个最小位置索引到辅助栈当中
*/

class minStack
{
public:
    vector<int> data;
    vector<int> minIndex;
    
    //push的时候判断当前元素和最小元素的关系。选择push到索引辅助栈的值
    void push(const int &elem)
    {
        data.push_back(elem);
        if(data.size() == 1)
        {
            minIndex.push_back(0);
        }
        else
        {
            int curMinIndex = minIndex[minIndex.size()-1];
            if(data[curMinIndex] < elem)
            {
                minIndex.push_back(curMinIndex);
            }
            else
            {
                minIndex.push_back(minIndex.size()-1);
            }
        }
    }
    
    //pop的时候，同时popminindex即可
    void pop()
    {
        if(data.size() > 0)
        {
            data.pop_back();
            minIndex.pop_back();
        }
    }
    
    //min操作直接返回data里的最小值
    int min()
    {
        return data[minIndex[minIndex.size()-1];
    }
};

int main(int argc, char *argv[])
{
    return 0;
}