题目:
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的min函数。
思路:
这个问题第一反应其实就是每次压入一个新元素进栈的时候就将栈里面的元素进行排序,让最小的元素放在栈顶,就能在o(1)的时间内得到最小元素。但如果这样的话就脱离了栈的本来意义了,栈的特点就在于先进后出,没有这个特点就不叫栈了。所以我们是不是可以想到在栈里添加一个成员变量用来存放最小的元素。每次压入一个新元素进栈的时候,如果该元素比当前最小的元素还要小,则更新最小元素。但是如果最小的元素已经弹出去了,我们这个时候又怎么找出此时的最小元素嘞,所以最关键的还是要时刻都在栈顶放的是最小元素就ok了,我么不一定只搞一个栈嘛,我们可以再设置一个辅助栈来帮助我们每次压元素进栈的时候比较压入的元素和要压的元素哪个小,把小的放到辅助栈里面,保证每一步压栈,辅助栈的栈顶都有最小值。
代码实现:
#define MAX_SIZE 10001
typedef struct {
int* data;
int top;
int min;
} MinStack;
MinStack* minStackCreate() {
MinStack* obj = (MinStack*)malloc(sizeof(MinStack));
obj->data = (int*)malloc(sizeof(int) * MAX_SIZE);
obj->top = -1;
obj->min = INT_MAX;
return obj;
}
void minStackPush(MinStack* obj, int x) {
if (obj->top >= MAX_SIZE) {
return;
}
obj->data[++(obj->top)] = x;
if (x < obj->min) {
obj->min = x;
}
}
void minStackPop(MinStack* obj) {
if (obj->top < 0) {
return;
}
obj->top--;
if (obj->top >= 0) {
obj->min = obj->data[obj->top];
} else {
obj->min = INT_MAX;
}
for (int i = obj->top; i >= 0; i--) {
if (obj->data[i] < obj->min) {
obj->min = obj->data[i];
}
}
}
int minStackTop(MinStack* obj) {
return obj->data[obj->top];
}
int minStackMin(MinStack* obj) {
return obj->min;
}
void minStackFree(MinStack* obj) {
free(obj);
}