定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
主要还是降低复杂度:
一方面:在最小栈的初始化中存入最大值,而不用判断存入的第一个值放入最小栈的第一个。
另一方面:直接利用最小栈的栈顶元素,不需要定义一个最小值记录。
stack<int> s; /*原始栈*/
stack<int> mins;/*记录最小值的栈*/
/*关键:最小栈与原始栈的记录同步,最小栈相同位置代表当前位置以前的最小值*/
int mini = 2147483647; /*int的最大值*/
/* 在最小栈的初始化函数里面*/
MinStack() {
mins.push(mini);
}
void push(int x) {
s.push(x);
mini = x>mins.top()?mins.top():x; /*找插入元素与栈内最小值谁更小*/
mins.push(mini);
}
void pop() {
s.pop();
mins.pop();
}
int top() {
return s.top();
}
int min() {
return mins.top();
}