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

来源:互联网 发布:talk软件下载 编辑:程序博客网 时间:2024/05/21 15:47

常见的思路可能会有:

(1)每入栈一个元素,对栈内元素进行排序,让最小的元素位于栈顶,但是这样不能保证栈的先进后出的特性,不符合题意

(2)设置一个成员变量始终代表最小的元素,每入栈一个元素,若比最小元素小,则更新。但当当前栈顶元素为最小的元素,其出栈后,栈中剩下元素的最小值则无法确定。
根据常见思路(2),可以进一步考虑,如果当某个元素入栈时,都能找出当前栈内元素的最小值存入另外一个栈(存放最小值的栈)中,那么当栈顶元素出栈时,

存放最小元素的栈也跟着出栈,那么此时存放最小元素的栈的栈顶元素即为当前栈内最小的元素。

import java.util.Stack;public class Solution {//栈1用于存放数据private Stack<Integer> stack1 = new Stack<>();//栈2用于存放当前数据栈内的最小元素private Stack<Integer> stack2 = new Stack<>();public void push(int node) {//将数据压入数据栈中stack1.push(node);if (stack2.isEmpty()) {//如果stack2中没有元素,则当前元素即为最小元素,压入stack2中stack2.push(node);} else {//如果stack2中存在元素,那么将当前元素与stack2栈顶元素比较,若比栈顶元素小,则将当前元素入栈,说明当前元素为stack1中最小的元素//若比stack2栈顶元素大或相等,则将stack2的栈顶元素再次入栈,作为当前stack1中的最小元素if (node < stack2.peek()) {stack2.push(node);} else {Integer peek = stack2.peek();stack2.push(peek);}}}public void pop() {if (!stack1.isEmpty() && !stack2.isEmpty()) {//stack1出栈的同时,stack2也必须同步出栈stack1.pop();stack2.pop();}}public int top() {Integer top = Integer.MIN_VALUE;if (!stack1.isEmpty()) {top = stack1.peek();}return top;}public int min() {if (!stack2.isEmpty()) {return stack2.peek();}return -Integer.MIN_VALUE;}}


 
阅读全文
0 0