【剑指offer】面试题30. 包含min函数的栈

2020.7.9

class MinStack {
public:
    /** initialize your data structure here. */
    MinStack() {

    }
    stack<int> minS, s;
    void push(int x) {
        s.push(x);
        if (minS.empty()) minS.push(x);//minS为空时，则压入x
        else {
            if (minS.top() > x) {//minS.top大于x，则把x压入
                minS.push(x);
            }
            else {
                minS.push(minS.top());
            }
        }
    }

    void pop() {
        s.pop();
        minS.pop();
    }

    int top() {
        return s.top();
    }

    int min() {
        return minS.top();
    }
};

---

解题思路

使用一个辅助栈，s_min，在每次s_data压入数据后，与s_min的top进行比较，如果比s_min当前的top的大，则将该数据压入到s_min中，如果s_min的top更大，则将top压入到s_min中

代码

class MinStack {
public:
	/** initialize your data structure here. */
	stack<int> s_data;//存放数据
	stack<int> s_min;//存放每一次压入后，栈内的最小值
	MinStack() {	
	}
	void push(int x) {
		s_data.push(x);
		if (s_min.size() == 0) s_min.push(x);
		else
		{
			if (x < s_min.top()) s_min.push(x);
			else s_min.push(s_min.top());
		}
	}
	void pop() {
		if (s_data.size() == 0) return;
		s_data.pop();
		s_min.pop();
	}
	int top() {
		return s_data.top();
	}
	int min() {
		return s_min.top();
	}
};