最小栈实现(简单的vector + 指针)

题目:

设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。

push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() —— 检索栈中的最小元素。
示例:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin();   --> 返回 -3.
minStack.pop();
minStack.top();      --> 返回 0.
minStack.getMin();   --> 返回 -2.

思路: 使用vector实现栈的push和pop功能,额外添加一个ptr指针始终指向栈的最小值;

class MinStack {
private:
    vector<int> data;
    int ptr;

public:
    /** initialize your data structure here. */
    MinStack() {
        ptr = 0;
    }
    void push(int x) {
        data.push_back(x);
        if(x < data[ptr]){
            ptr = data.size() - 1;
        }
    }
    void pop() {
        if(ptr == data.size() - 1){
            ptr = 0;
            for(int i = 1; i < data.size() - 1;i++)
            {
                if(data[i] <= data[ptr]){
                    ptr = i;
                }
            }
        }
        data.pop_back();
    }
    int top() {
        return data[data.size() - 1];
    }
    int getMin() {
        return data[ptr];
    }
};

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack* obj = new MinStack();
 * obj->push(x);
 * obj->pop();
 * int param_3 = obj->top();
 * int param_4 = obj->getMin();
 */

你可能感兴趣的:(algorithm,practice)