您的位置:首页 > 职场人生

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

2017-09-04 14:53 330 查看
常见的思路可能会有:

(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;

}

}




                                            
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息