21.包含min函数的栈

题目

定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。

思路

用一个栈data保存数据,用另外一个栈min保存入栈最小的数,在压入时比较当前元素是否小于min栈顶的元素,小于则min压入当前元素,否则再次压入min栈顶的元素,即始终保持min栈顶是当前最小值,同样在data弹出的时候min也要弹出。

代码

package com.zhumq.leetcode;
import java.util.Stack;

public class MinStack {
	//存储数据
	Stack<Integer> date = new Stack<>();
	//存储最小值
    Stack<Integer> min = new Stack<>();
    public void push(int node) {
        date.push(node);
        if (min.empty()) {
            min.push(node);
        } else {
        	//min栈顶始终是最小值
        	min.push((min.peek()) > node ? node : min.peek());
        }
    }

    public void pop() {
        date.pop();
        min.pop();
    }

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

    public int min() {
        return min.peek();
    }
}

猜你喜欢

转载自blog.csdn.net/DjokerMax/article/details/82969180