题目
leetcode 155
设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) – 将元素 x 推入栈中。
pop() – 删除栈顶的元素。
top() – 获取栈顶元素。
getMin() – 检索栈中的最小元素。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.
代码
用链表实现栈,采用头插法,每次存储数据+当前所有数据最小值。
typedef struct Stack{
int data, mindata;
struct Stack *next;
} MinStack;
/** initialize your data structure here. */
MinStack* minStackCreate() {
MinStack *obj = (MinStack *)malloc(sizeof(MinStack));
obj->next = NULL;
return obj;
}
void minStackPush(MinStack* obj, int x) {
MinStack *stack = minStackCreate();
stack->data = x;
if (obj->next == NULL || obj->next->mindata >= x) {
stack->mindata = x;
} else {
stack->mindata = obj->next->mindata;
}
stack->next = obj->next;
obj->next = stack;
}
void minStackPop(MinStack* obj) {
obj->next = obj->next->next;
}
int minStackTop(MinStack* obj) {
return obj->next->data;
}
int minStackGetMin(MinStack* obj) {
return obj->next->mindata;
}
void minStackFree(MinStack* obj) {
if (obj == NULL) return ;
minStackFree(obj->next);
free(obj);
}