155.最小栈-力扣(LeetCode)

目录

  • 55.最小栈-力扣(LeetCode)
  • 正确题解
  • 优化题解1
  • 优化题解2

55.最小栈-力扣(LeetCode)

155.最小栈-力扣(LeetCode)_第1张图片
155.最小栈-力扣(LeetCode)

题目描述:设计push、pop、top、getMin,函数,并且可以在时间复杂度为O(1)的情况下找出最小值
错误的思路:给定一个变量,来记录最小值,这种想法是错误的,我们应该要更全面的考虑问题。
当我们入栈{6,5,4,3,2,1}时,每次记录最小值的变量随着每次入栈而改变,当数据全部入栈,记录最小值的变量就为1,如果在不删除数据的情况下,可以以O(1)的时间复杂度找到最小值,但是如果删除出来一个栈顶的元素,则栈中的最小值也发生了改变,与记录最小值的变量不匹配。
155.最小栈-力扣(LeetCode)_第2张图片

正确题解

通过上面的错误思路,我们可以加以改进。
在每一次元素入栈时,记录当前栈中的最小值,存放在另一个栈中。
例如:{6,5,4,3,2,1}
当6入栈时,当前栈中的元素最小值为6;
当5入栈时,当前栈中的元素最小值为5;
当4入栈时,当前栈中的元素最小值为4;
当3入栈时,当前栈中的元素最小值为3;
当2入栈时,当前栈中的元素最小值为2;
当1入栈时,当前栈中的元素最小值为1;
把每次入栈时的最小元素存放在一个栈(min_stack)中来保存。
当删除栈顶的元素时,min_stack也随之删除栈顶元素,保证了栈中元素的最小值与min_stack对应,找栈中最小值时直接在min_stack中的栈顶拿。时间复杂度为O(1),这种做法叫以空间换时间
155.最小栈-力扣(LeetCode)_第3张图片
代码

class MinStack {
public:
    /** initialize your data structure here. */
    MinStack() {}
    
    void push(int val) {
        st.push(val);
        if(min_stack.empty()||val<=min_stack.top())
        min_stack.push(val);
        else
        {
            min_stack.push(min_stack.top());
        }
    }
    
    void pop() {     
        st.pop();
        min_stack.pop();
    }
    
    int top() {
        return st.top();
    }
    
    int getMin() {
        return min_stack.top();
    }
    private:
        stack<int> st;
        stack<int> min_stack;
};

优化题解1

在上面的方法中,存在着空间浪费。
例如:{5,3,2,4,6,1}
在min_stack中也存放{5,3,2,2,2,1},在入栈4,6时,也要在min_stack中存放2,形成了一定程度的空间浪费。
有什么好解决的方法吗?如果呢避免就好了。
当入栈的元素比min_stack栈顶的元素大时,min_stack则不需要保存这时栈中的最小值。
只有入栈的元素比min_stack栈顶的元素小或者等于时,才需要保存这时栈中的最小值。
例如:{5,3,2,4,6,1}
1:5入栈时,当前最小值为5,min_stac保存5,这时min_stack中只要5:{5}
2:3入栈时,当前最小值为3,min_stac保存3,这时min_stack中只要5:{5,3}
3:2入栈时,当前最小值为2,min_stac保存2,这时min_stack中只要5:{5,3,2}
4:4入栈时,当前最小值为2,不需要保存这时的最小值,这时min_stack中只要5:{5,3,2}
5:6入栈时,当前最小值为2,不需要保存这时的最小值,这时min_stack中只要5:{5,3,2}
6:1入栈时,当前最小值为1,min_stac保存1,这时min_stack中只要5:{5,3,2,1}
在删除栈顶元素时,当栈顶元素与min_stack栈顶元素相同时,则min_stack栈顶元素也删除。不同时,min_stack栈顶元素不需要删除。
155.最小栈-力扣(LeetCode)_第4张图片
代码:

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

    MinStack() {}
    
    void push(int val) {
        st.push(val);
        if(min_stack.empty()||val<=min_stack.top())
        min_stack.push(val);
    }
    
    void pop() {
        if(st.top()==min_stack.top())
        {
            min_stack.pop();
        }
        st.pop();
    }
    
    int top() {
        return st.top();
    }
    
    int getMin() {
        return min_stack.top();
    }
    private:
    	stack<int> st;
    	stack<int> min_stack;
};

优化题解2

在优化题解1中还会存在浪费空间。
例如:{5,3,2,2,2,6,4,1,1,1}
用优化题解1中的方法:min_stack:{5,3,2,2,2,1,1,1},出现了重复的2和1。
如果出现极端情况,出现了N个2,那么在min_stack中也要存入N个2,浪费了大量的空间。
有什么方法解决这个问题吗?
计数方法:和计数排序一样的原理。
例如:{5,3,2,2,2,6,4,1,1,1}
用一个结构体来记录:

struct val_count
{
	int _val;//记录最小值
	int _count://记录次数
};

在min_stack:{{5,1},{3,1},{2,3},{1,3}}——>前一个数字表示这个阶段的最小值,后面的数字表示这个阶段最小值出现的次数。
155.最小栈-力扣(LeetCode)_第5张图片
直到_count减到0,才删除min_stack的栈顶元素。

代码

struct val_count
{
    int _val;
    int _count;
};
class MinStack {
public:
    /** initialize your data structure here. */
    
    MinStack() {}
    
    void push(int val) {
        st.push(val);
        if(min_stack.empty()||val<(min_stack.top()._val))
        {
           vat_count temp={val,1};
           min_stack.push(temp);
        }
        else
        {
            if(val==(min_stack.top()._val))
            min_stack.top()._count++;
        }
    }
    
    void pop() {
        if(st.top()==min_stack.top()._val)
        {
            min_stack.top()._count--;
            if(min_stack.top()._count==0)
            min_stack.pop();
        }
        st.pop();
    }
    
    int top() {
        return st.top();
    }
    
    int getMin() {
        return min_stack.top()._val;
    }

    private:
        stack<int> st;
        //stack min_stack;
        stack<vat_count> min_stack;
};

到此结束啦

你可能感兴趣的:(每日一题,数据结构,leetcode,算法,python)