一、题目
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.
提示:
各函数的调用总次数不超过 20000 次
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/bao-han-minhan-shu-de-zhan-lcof
二、思路
我的思考过程:栈有两种实现方式,一种是链表,一种是数组
我的第一反应是数组,因为数组有随机存取的特性,可以在用一个变量存储当前最小的值,但是如果用数组来实现栈,在push 的时候,每次都需要重新开辟内存空间然后把之前的节点全部转移到新的空间去,这样在push的时候时间复杂度不可能为O(1),所以只能用链表实现。
使用链表的时候,push和pop的时间复杂度都可以保证为O(1),那么问题就到了如何找到最小的元素并且使得时间复杂度为O(1)。我一开始想了很多的方法,但是时间复杂度都不能保证,后来看了一下评论区,某大佬说可以在节点里面开辟一个空间专门用来存储当前栈的最小值,我茅塞顿开,于是开始用链表实现栈。选取带头节点的单链表来实现。下面附上完整代码。
三、C语言解法
typedef struct {
int data;//数据域
struct MinStack *next;//指向下一个节点的指针
int min;//对于以当前节点为顶的栈的最小元素
} MinStack;
/** initialize your data structure here. */
MinStack* minStackCreate() {
MinStack *p = (MinStack*)malloc(sizeof(MinStack));//初始化头结点
p->next = NULL;//头结点的next指向NULL
return p;//返回头结点的指针
}
void minStackPush(MinStack* obj, int x) {
MinStack *q = (MinStack*)malloc(sizeof(MinStack));//申请一个结点的空间
MinStack *p = obj->next;//获取指向obj的next域的指针,防止丢失指针
q->data = x;//新节点的数据域赋值为x
if(p==NULL||q->data<p->min)//obj为空或者当前栈的最小元素小于x时
q->min = x;//当前栈(包含了新加入的结点)的最小元素就为x
else q->min = p->min;//不满足条件则说明最小值不变
q->next=obj->next;//结点入栈
obj->next=q;
}
void minStackPop(MinStack* obj) {
MinStack *q = obj->next;//头结点的next结点就是栈顶
obj->next= q->next;//头结点指向next结点的next结点
free(q);//释放空间
}
int minStackTop(MinStack* obj) {
MinStack *q = obj->next;//头结点的next结点就是栈顶
return q->data;//返回栈顶data域
}
int minStackMin(MinStack* obj) {
MinStack *q = obj->next;//头结点的next结点就是栈顶
return q->min;//返回栈顶min域
}
void minStackFree(MinStack* obj) {
while(obj->next!= NULL)//循环删除栈,直到只剩下头结点
free(obj);//释放头结点的空间
}
/**
* Your MinStack struct will be instantiated and called as such:
* MinStack* obj = minStackCreate();
* minStackPush(obj, x);
* minStackPop(obj);
* int param_3 = minStackTop(obj);
* int param_4 = minStackMin(obj);
* minStackFree(obj);
*/