剑指offer 包含min函数的栈

题目描述

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

先介绍一下java.util.Stack.peek()方法实例：

下面的例子显示java.util.Stack.peek()方法的使用

package com.yiibai;

import java.util.*;

public class StackDemo {
public static void main(String args[]) {
// creating stack
Stack st = new Stack();

// populating stack
st.push("Java");
st.push("Source");
st.push("code");

// checking the top object
System.out.println("Top object is: "+st.peek());
}
}

现在编译和运行上面的代码示例，将产生以下结果。

Top object is: code

本题解决方案：

import java.util.Stack;

/**
*
*/
public class Solution {

Stack<Integer> dataStack = new Stack<Integer>();
Stack<Integer> minStack = new Stack<Integer>();

public void push(int node) {
dataStack.push(node);
if(minStack.isEmpty() || node < minStack.peek()){
minStack.push(node);
}
else{
minStack.push(minStack.peek());

}
}

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

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

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