您的位置:首页 > Web前端

剑指offer题21 包含min函数的栈

2015-12-06 20:48 459 查看

题目描述

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

题目解析:
题目的重点在于,每次调用min函数都可以pop出当前栈中的最小值;
可以考虑使用一个辅助栈minStack,当每次push进来一个数值node时,minStack将当前栈顶值(默认当前最小值)与node进行比较,push捡来最小值;则在每个时刻,minStack中保存的都是当前栈中的最小值;然后按顺序pop出来就行了;(要注意pop时,数据栈和辅助栈中的数据要保持同步,因此应该同时pop)

代码:

public class Solution {
private Stack<Integer> dataStack = new Stack<>(); // 数据Stack
private Stack<Integer> minStack  = new Stack<>(); // 保存最小值的辅助Stack

public void push(int node) {
dataStack.push(node);

if (minStack.isEmpty())
minStack.push(node);
else
minStack.push(Math.min(node, minStack.peek()));
}

public void pop() {
dataStack.pop();
minStack.pop();
}

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

public int min() {
return minStack.peek();
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: