leetcode题目分析(一)leetcode155最小栈

一、前言

本题基于leetcode155最小栈这道题,说一下通过java解决的一些方法。
需要尤其注意的是,此题输入的值的区间范围在-2^31 <= val <= 2^31 - 1.这将会影响我们最后一种最优解的结果出现问题。这些都是后话。

二、解决思路

其实在一开始的提交记录,我的方案忽略了题干中的常数时间,而是使用了偏向于工程的,将栈的所有元素放到数组中,然后通过数组的stream流的min方法解决:

class MinStack {

private Stack<Integer> stack;
	public MinStack() {
		stack = new Stack<>();
	}

	public void push(int val) {
		stack.push(val);
	}

	public void pop() {
		stack.pop();
	}

	public int top() {
		return stack.peek();
	}

	public int getMin() {
		ArrayList<Integer> arrayList = new ArrayList<>();
		arrayList.addAll(stack.subList(0, stack.size()));
		return arrayList.stream().min(Integer::compare).get();
	}
}

leetcode题目分析(一)leetcode155最小栈_第1张图片

那么实际上stream也是一个循环,但是结果过了,可能并没有对时间去做一个检测。但是这个解决方案时间复杂度和空间复杂度均为O(n),是最拉的方案。所以这里完全不推荐。那么下面我们说一下一些比较OK的正解

2.1、辅助栈

这也是最容易想到的一种解决方案,我们定义一个辅助栈。当栈为空的时候,对于主栈和辅助栈都存放该元素。如果不为空,每次push的时候,通过将当前要插入的元素和辅助栈的栈顶元素去做一个对比,如果要插入的元素比栈顶元素小,则push要插入的元素,反之push辅助栈的栈顶元素。这样我们可以保证我们的辅助栈的栈顶一定对应的最小值,而出栈查看栈顶判空都很简单:出栈两个栈都需要pop,查看栈顶返回我们主栈的栈顶元素,查看最小值返回我们辅助栈的栈顶元素即可。

class MinStack {

	private Stack<Integer> stack;
	private Stack<Integer> assistStack;

	public MinStack() {
		stack = new Stack<>();
		assistStack = new Stack<>();
	}

	public void push(int val) {
		if(stack.isEmpty()){
			stack.push(val);
			assistStack.push(val);
		}else {
			stack.push(val);
			assistStack.push(Math.min(assistStack.peek(), val));
		}
	}

	public void pop() {
		stack.pop();
		assistStack.pop();

	}

	public int top() {
		return stack.peek();
	}

	public int getMin() {
		return assistStack.peek();
	}
}

2.2、链表

我们可以通过自定义链表的方式,去定义一个节点,节点中增加一个属性min记录最小值,然后每次push的时候,类似链表的增加元素,而对于新的Node节点的min属性,思路很像上面辅助栈的思路,用要插入的值和head节点(其实就类似栈顶元素)的min做一个对比,如果要比min还小,则这个新节点的min就是要插入的值,反之是head节点的min.然后剩下的就是链表实现栈的常规操作了。代码如下:

class MinStack {

private Node head;
    
    public void push(int x) {
        if(head == null) 
            head = new Node(x, x);
        else
            //用当前要插入的值和头结点的最小值做一个对比
            head = new Node(x, Math.min(x, head.min), head);
    }

    public void pop() {
        head = head.next;
    }

    public int top() {
        return head.val;
    }

    public int getMin() {
        return head.min;
    }
    
    private class Node {
        int val;
        int min;
        Node next;
        
        private Node(int val, int min) {
            this(val, min, null);
        }
        
        private Node(int val, int min, Node next) {
            this.val = val;
            this.min = min;
            this.next = next;
        }
    }
}

上面两个思路都差不多,所以最终的解决时间空间也都差不多,时间复杂度是O(1),空间复杂度为O(n)
leetcode题目分析(一)leetcode155最小栈_第2张图片

2.3、存差值

那有没有时间复杂度和空间复杂度均为O(1)的解决方案呢,其实是有的。
我们可以额外存储一个值记录当前最小值min,然后我们的栈不放常规元素了。放差值,这个差值就是要插入元素x和我记录的最小值min的差值。记diff = x-min,这样对于getMin()操作我就可以直接返回记录的min了
对于push操作
当栈为空栈时,此时min就是要插入的值,diff记为0,所以push(0),记min = x;
当栈不为空,此时计算diff
如果diff >= 0; 那么说明要插入的值>=最小值,此时我们push(diff),但是我的min是不需要变动的
如果diff < 0:那么说明要插入的值<最小值,此时push(diff),但是我们记录min = x;

对于pop操作
由于我的最小值是一直在变动的,所以我们仍然需要查看栈顶即diff的值
如果diff<0的出栈了,这说明此时我的一个最小值出栈了,那么此时我需要变动我的min为min = min-diff
例如我的diff = -1,我的min记录此时为-3,那么我的min应该变成-3-(-1) = -2;
如果diff >=0的出栈,那就出栈,不用变动

对于top操作
主要在于还原值
此时查看栈顶的diff值
如果diff<0,那么肯定当前栈顶的元素就是最小值,那么直接返回最小值min即可
如果diff>0,那么说明原来的元素要比最小值大diff,所以我应该返回min+diff;
那么代码如下:

class MinStack {

	private Stack<Integer> stack;
	private Integer min;

	public MinStack(){
		stack = new Stack<>();
		min = 0;
	}
	public void push(int x) {
		if(stack.isEmpty()){
			//一开始由于栈是空栈,所以我们放入0.diff=0
			stack.push(0);
			min = x;
		}else {
			//不为空时,我们存插入的值和最小值的差值
			int diff = x - min;
			stack.push(diff);
			//如果差值<0,说明要插入的比当前最小值还要小,此时更新最小值
			if(diff < 0){
				min = x;
			}
		}
	}

	public void pop() {
		Integer pop = stack.pop();
		//如果
		if(pop < 0){
			min = min - pop;
		}
	}

	public int top() {
		Integer diff = stack.peek();
		if(diff < 0){
			return min;
		}else {
			return min + diff;
		}
	}

	public int getMin() {
		return min;
	}
}

但是需要注意的是:如果没有进行数值范围限制,上面的方法能行吗?答是不行,因为数值没有限制的话,差值的计算可能会溢出。例如Leetcode本题就无法用这个方法,因为这个方法一开始的数值范围是-2^31 <= val <= 2^31 - 1.那么在pop计算min-diff会出现下面的一幕,:

leetcode题目分析(一)leetcode155最小栈_第3张图片
不过我们需要掌握这种思路,这种思路的时空复杂度均为O(1).也是最小栈的最优解

你可能感兴趣的:(leetcode题目分析,leetcode,算法,java,数据结构,职场和发展)