welcome to my blog
剑指offer面试题30(java版):包含min函数的栈
题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))
第四次做; 核心: 如何维护最小值? stack2的栈顶元素是当前的最小值; 如果x比stack2栈顶元素小,就压入x; 如果x比stack2栈顶元素大,就压入stack2的栈顶元素
class MinStack {
private int min;
private Stack<Integer> stack1;
private Stack<Integer> stack2;
/** initialize your data structure here. */
public MinStack() {
min = Integer.MAX_VALUE;
stack1 = new Stack<>();
stack2 = new Stack<>();
}
public void push(int x) {
stack1.push(x);
//这么写是错误的, 因为如果stack2空了, 此时的min是历史的min, 而此时应该重置min
// if(x<min)
// min = x;
// stack2.push(min);
//这样写也不对, 如果stack2不为空, 但是min被弹出了, 此时的min并没有改变, 而正确的min应该是stack2栈顶元素!
// if(stack2.isEmpty())
// min = x;
// if(x < min)
// min = x;
// stack2.push(min);
if(stack2.isEmpty()){
stack2.push(x);
}
else{
stack2.push(x<stack2.peek()? x:stack2.peek());
}
}
public void pop() {
stack1.pop();
stack2.pop();
}
public int top() {
return stack1.peek();
}
public int min() {
return stack2.peek();
}
}
/**
* Your MinStack object will be instantiated and called as such:
* MinStack obj = new MinStack();
* obj.push(x);
* obj.pop();
* int param_3 = obj.top();
* int param_4 = obj.min();
*/
笔记
- 声明并初始化栈 Stack s = new Stack(); 一般都要加上泛型, 要不默认元素类型为Object
- s.pedk():查看栈顶元素但并不弹出
思路
- 创建一个辅助栈, 只存放主栈中的最小值, 举个例子
- 主栈: 3,4,5,5,5,6,1,2,7
- 辅助栈:3,3,3,3,3,3,1,1,1
第三次做, s2并不是每次都push, 减少了push操作的流程, 但是会增加pop操作的流程
import java.util.Stack;
public class Solution {
Stack<Integer> s1 = new Stack<Integer>();
Stack<Integer> s2 = new Stack<Integer>();
int min = Integer.MAX_VALUE;
public void push(int node) {
s1.push(node);
if(node<min){
s2.push(node);
min = node;
}
}
public void pop() {
if(s1.pop()==s2.peek())
s2.pop();
}
public int top() {
return s1.peek();
}
public int min() {
return s2.peek();
}
}
第二次做, 好好使用辅助栈s2
import java.util.Stack;
public class Solution {
/*
使用两个栈实现
s1正常压入
s2为空时直接压入
s2中有元素时,取出栈顶元素,如果待压入的元素小于栈顶元素则压入该元素;如果待压入的元素大于等于栈顶元素则在压入一次栈顶元素
改写一下常规的push和pop即可
*/
Stack<Integer> s1 = new Stack<>();
Stack<Integer> s2 = new Stack<>();
public void push(int node) {
s1.push(node);
if(s2.isEmpty())
s2.push(node);
else{
int temp = node < s2.peek()? node:s2.peek();
s2.push(temp);
}
}
public void pop() {
s1.pop();
s2.pop();
}
public int top() {
return s1.peek();
}
public int min() {
return s2.peek();
}
}
第二次做, 可以拿个全局变量记录最小值, 不用每次都访问s2的栈顶
import java.util.Stack;
public class Solution {
/*
使用两个栈实现
s1正常压入
s2为空时直接压入
s2中有元素时,取出栈顶元素,如果待压入的元素小于栈顶元素则压入该元素;如果待压入的元素大于等于栈顶元素则在压入一次栈顶元素
改写一下常规的push和pop即可
*/
Stack<Integer> s1 = new Stack<>();
Stack<Integer> s2 = new Stack<>();
int min = Integer.MAX_VALUE;
public void push(int node) {
s1.push(node);
if(node < min)
min = node;
s2.push(min);
}
public void pop() {
s1.pop();
s2.pop();
}
public int top() {
return s1.peek();
}
public int min() {
return s2.peek();
}
}
import java.util.Stack;
public class Solution {
Stack<Integer> s = new Stack<Integer>();
Stack<Integer> assistS = new Stack<Integer>();
int minV = Integer.MAX_VALUE;
public void push(int node) {
s.push(node);
if(minV > node)
minV = node;
assistS.push(minV);
}
public void pop() {
s.pop();
assistS.pop();
}
public int top() {
return s.pop();
}
public int min() {
return assistS.peek();
}
}