剑指offer20_包含min函数的栈

题目描述:

定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。

解决思路:

最开始的解题思路以为要用基本数据结构实现栈,后来发现用数组的话实在是复杂度太高了。

查了一下网上的思路才发现是用基本的栈去实现复杂的栈。

具体思路是维护两个栈,其中一个是基本的栈,另一个保存最小元素的栈。

注意:一定要保持两个栈的同步,pop的时候都pop,push的时候都push,这样才能保证min栈的首元素是当前最小值。

具体代码如下:

import java.util.Stack;

public class Solution {
    Stack<Integer> stack=new Stack<Integer>();//存放每次新来的元素
    Stack<Integer> minStack=new Stack<Integer>();//以顶元素最小的方式存放当前最小值
    public void push(int node) {
        stack.push(node);
        if(minStack.isEmpty()||node<minStack.peek()){ //一定要考虑minstack为空的情况
            minStack.push(node);
        }else{
            minStack.push(minStack.peek());
        }
    }
    
    public void pop() {
        stack.pop();
        minStack.pop();
    }
    
    public int top() {
        return stack.peek();
    }
    
    public int min() {
        return minStack.peek();
    }
}
 

猜你喜欢

转载自blog.csdn.net/lilililililydia/article/details/88546858