数据结构复习整理——栈的基本操作

#include<cstdio>
#include<cstdlib>
#define STACK_INIT_SIZE 100
#define STACKINCREMENT 10
#define ERROR 0
#define OK 1
typedef struct {
    int *base;
    int *top;
    int stacksize;
}Sqstack;


//建立栈
int Initstack(Sqstack &S)
{
    S.base = (int *)malloc(STACK_INIT_SIZE*sizeof(int));
    if(!S.base)
        return ERROR;
    S.top = S.base;
    S.stacksize = STACK_INIT_SIZE;
    return OK;
}


//向栈中插入元素
int Push(Sqstack &S,int e)
{
    if(S.top-S.base>S.stacksize){
        S.base =(int *)realloc(S.base,S.stacksize+STACKINCREMENT*sizeof(int));
        if(!S.base)
            return ERROR;
        S.top=S.base+S.stacksize;
        S.stacksize+=STACKINCREMENT;
    }
    *S.top++=e; //这里先取值然后再加一，原因是使栈底也有元素
    return OK;
}


//从栈中删除元素
int Pop(Sqstack &S)
{
    int m;
    if(S.top == S.base)
        return ERROR;
    m=*--S.top; //这里先减一然后再取值，原因是此时栈顶没有元素
    return m;
}


int main()
{
    Sqstack S;
    Initstack(S);
    int n,m;
    scanf("%d",&n);
    int a[n];
    for(int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
        Push(S,a[i]);
    }
    for(int i=0;i<n;i++)
    {
        m=Pop(S);
        printf("%d ",m);
    }
    return 0;


}