【剑指 Offer 学习】【面试题 21： 包含 min 函数的栈】【思路】

题目：

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

 

思路：

用一个栈 data 保存数据，用另外一个栈 min 保存依次入栈最小的数 

比如，data中依次入栈，5, 4, 3,   8, 10, 11, 12, 1

          则 min 依次入栈，5, 4, 3，no,no, no, no, 1

no 代表此次不如栈 每次入栈的时候，如果入栈的元素比 min 中的栈顶元素小或等于则入栈，否则不如栈。

弹栈时，数据栈依次弹出，当最小值栈栈顶元素与数据栈栈顶元素相同时，最小值栈弹栈。

 

代码：

package 包含min函数的栈;

import java.util.Stack;

/**
 * 定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。
 * 思路：用一个栈 data 保存数据，用另外一个栈 min 保存依次入栈最小的数 
 * 比如，data中依次入栈，5, 4, 3, 8, 10, 11, 12, 1
 *      则 min 依次入栈，5, 4, 3，no,no, no, no, 1
 * 
 * no 代表此次不如栈 每次入栈的时候，如果入栈的元素比 min 中的栈顶元素小或等于则入栈，否则不如栈。
 * 
 * @author Administrator
 *
 */
public class Demo {
	// 数据栈，用于存放插入的数据
	Stack<Integer> data = new Stack<>();
	// 最小数栈，存放数据栈中最小的数的位置
	Stack<Integer> min = new Stack<>();
	Integer temp = null;

	public void push(int node) {
		// 数据栈存放插入数据
		data.push(node);
		// 最小数栈为空时存放第一个插入的数据
		if (min.isEmpty()) {
			min.push(node);
		} else {
			// 插入数据小于最小数栈的栈顶元素时，插入最小数栈
			if (min.peek() > node)
				min.push(node);
		}
	}

	public void pop() {
		// 数据栈与最小值栈的栈顶元素相等时，此栈顶元素为最小值
		if (data.pop() == min.peek()) {
			min.pop();
		}
	}

	public int top() {
		// 返回数据栈的栈顶元素
		return data.peek();
	}

	public int min() {
		// 返回最小值栈的栈顶元素，即最小值
		return min.peek();
	}

	public static void main(String[] args) {
		Demo demo = new Demo();
		demo.push(3);
		System.out.println(demo.min() == 3);
		demo.push(4);
        System.out.println(demo.min() == 3);
        demo.push(2);
        System.out.println(demo.min() == 2);
        demo.push(3);
        System.out.println(demo.min() == 2);
        demo.pop();
        System.out.println(demo.min() == 2);
        demo.pop();
        System.out.println(demo.min() == 3);
        demo.push(0);
        System.out.println(demo.min() == 0);
	}
}