一、题目

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

二、思路

用一个栈dataStack保存数据,用另外一个栈minStack保存依次入栈最小的数。每次元素存入minStack的时候,如果该元素比minStack的栈顶元素小,则存入minStack,否则用minStack栈顶元素代替该元素存入minStack。

比如dataStack中依次入栈的元素为:5,  4,  3,  8,  10,  11,  12,  1

则minStack依次入栈的元素为:      5,  4,  3,  3,   3,   3,    3,    1

三、代码

import java.util.Stack;

/*
用一个栈dataStack保存数据,用另外一个栈minStack保存依次入栈最小的数。每次元素存入minStack的时候,如果该元素比minStack的栈顶元素小,则存入minStack,否则用minStack栈顶元素代替该元素存入minStack。
比如dataStack中依次入栈的元素为:5,4,3,8,10,11,12,1
则minStack依次入栈的元素为:5,4,3,3,3,3,3,1
*/
public class Solution {
Stack<Integer> dataStack = new Stack<Integer>(); //保存元素
Stack<Integer> minStack = new Stack<Integer>(); //保存最小元素 public void push(int value) {
dataStack.push(value); //元素入dataStack
if (minStack.isEmpty() || value < minStack.peek()) { //如果minStack为空,或者当前存入的元素小于minStack的栈顶元素,则把该元素存入minStack
minStack.push(value);
} else {
minStack.push(minStack.peek());//如果minStack不为空且当前存入的元素大于或者等于minStack的栈顶元素,则把minStack栈顶元素存入minStack
}
} public void pop() {
dataStack.pop();
minStack.pop();
} public int top() {
return dataStack.peek();
} public int min() {
return minStack.peek();
}
}

--------------------------------------------------------------------------------

参考链接:https://www.nowcoder.com/questionTerminal/4c776177d2c04c2494f2555c9fcc1e49

04-13 21:34