要求:设计一个栈的数据结构,含有min()操作来返回当前栈的最小值,要求min(),push
,pop的时间复杂度均为O(1).
主栈记录元素,使用一个辅助栈来记录最小值。
1.入栈:只要压入的值小于辅助栈栈顶元素,则同时将其压入辅助栈。
2.出栈:当主栈栈顶元素与辅助栈栈顶元素相等时,这说明最小元素出栈,所以辅助栈也随之出栈,其他情况仅对主栈操作就行
3.min:直接获得辅助栈栈顶元素,即当前主栈中的最小元素。
另一种实现方式是:采用结构体,每次压入元素及最小值,此时的空间复杂度高于上一种实现方式。
代码:
#include<stdio.h>
#define Slen 100
#define Elen 10
int stack[Slen];
int mstack[Slen];
int top = 0, mtop = 0;
void push(int a) {
stack[top++] = a;
if (mtop == 0) mstack[mtop++] = a;
else if (a < mstack[mtop-1]) mstack[mtop++] = a;
}
int pop() {
if (stack[top - 1] == mstack[mtop - 1])
mtop--;
return stack[--top];
}
int min() {
return mstack[mtop - 1];
}
int main(void)
{
int a[10] = { 4,3,2,1,2,3,5,6,3,9 };
for (int i = 0; i < 10; i++)
push(a[i]);
for (int i = 0; i < 7; i++)
{
int b = pop();
printf("%d ", b);
}
printf("\n%d", min());
}