stack.cpp

自己写的数据结构代码

#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include <malloc.h>
#include <process.h>

#define OK 1
#define ERROR -1
typedef int SElemType;
typedef int Status;
#define STACKINCREMENT 2

struct SqStack
{
	SElemType *base;
	SElemType *top;
	int stacksize;
};

int n;

Status InitStack(SqStack &S)
{
	S.base=(SElemType *)malloc((n+1)*sizeof(SElemType));
	
	S.top=S.base;
	S.stacksize=n;
	return OK;
}

Status Push(SqStack &S,SElemType e)
{
	if(S.top-S.base>=S.stacksize)
	{
		S.base=(SElemType *)realloc(S.base,(S.stacksize+STACKINCREMENT)*sizeof(SElemType));
		if(!S.base) exit(OVERFLOW);
		S.top=S.base+S.stacksize;
		S.stacksize=S.stacksize+STACKINCREMENT;
	}
	*S.top=e;
	S.top++;
	return OK;
}


Status StackTraverse(SqStack S)
{
	if(S.base==S.top) return ERROR;
	while(1)
	{
		printf("%d ",*(S.top-1));
		S.top--;
		if(S.base>=S.top)  break;
	}
	printf("\n");
	return OK;
}

int main()
{
	int j,w;
	SqStack s;
	scanf("%d",&n);
	if(InitStack(s)==OK)
	{
		for(j=1;j<=n;j++)
		{
			scanf("%d",&w);
			Push(s,w);
		}
		StackTraverse(s);
	}
	return 0;
}