题目:定义栈的数据结构,要求添加一个min函数,能够得到栈的最小元素。要求函数min、push以及pop的时间复杂度都是O(1)。
开始的时候觉得蛮简单,只要设置一个变量保存最小的那个值不就行了,不过程序运行后就发现了个问题,如果把最小的那个值pop出来了怎么办。后来在网上找资料说一种普遍的方法就是加一个辅助栈,存入最小值的位置,没错pop一个元素出栈时,也pop辅助栈。
下面就直接贴代码
#include <iostream>
#include <cstdio>
#include <cstring>
const int stackSize = 1001;
using namespace std;
typedef struct{
int data[stackSize];
int top;
int min;
}Stack;
Stack sk;
void init(Stack *s)
{
s->top = -1;
s->min = -1;
sk.top = -1;
}
void push(Stack *s, int val)
{
if(full(s))
{
cout << "stack is full." << endl;
return;
}
if((s->min == -1) || (val < s->min))
{
s->min = val;
sk.data[++sk.top] = val;
}
s->data[++s->top] = val;
}
int pop(Stack *s)
{
int val;
if(empty(s))
{
cout << "stack is empty." << endl;
return 0;
}
val = s->data[s->top--];
if(val == sk.data[sk.top])
{
s->min = sk.data[--sk.top];
}
return val;
}
bool full(Stack *s)
{
if(s->top == stackSize - 1)
return true;
return false;
}
bool empty(Stack *s)
{
if(s->top == -1)
return true;
return false;
}
int min(Stack *s)
{
return s->min;
}