数据结构——栈

一、栈的概念和结构

栈：一种特殊的线性表，其只允许在固定的一端进行插入和删除元素操作，进行数据插入和删除操作的一端称为栈顶，另一端称为栈底。栈中的数据元素遵守后进先出LIFO（Last In First Out）的原则。

压栈：栈的插入操作叫做进栈/压栈/入栈，入数据在栈顶。

出栈：栈的删除操作叫做出栈，出数据也在栈顶。

二、栈的实现

栈可以用链表和数组实现，本文用数组来实现栈，即数组栈。实现数组栈，要用数组的尾部当栈顶。

2.1栈的结构

typedef int STDataType;
typedef struct Stack
{
	STDataType* a;
	int top;
	int capacity;
}ST;

2.2栈的接口

//初始化栈
void STInit(ST* ps);
//栈的销毁
void STDestroy(ST* ps);
//入栈
void STPush(ST* ps, STDataType x);
//出栈
void STPop(ST* ps);
//获取栈的有效数据个数
int STSize(ST* ps);
//判断栈是否为空
bool STEmpty(ST* ps);
//访问栈顶元素
STDataType STTop(ST* ps);

2.3栈的初始化

需要注意的是，top可以初始化为-1，也可以初始化为0。若初始为-1，则top指向栈顶元素；若初始化为0，则top指向栈顶元素的下一个位置。这里我们将top初始化为0。

void STInit(ST* ps)
{
	assert(ps);
    //先给栈开辟一块空间
	ps->a = (STDataType*)malloc(sizeof(STDataType) * 4);
	if (ps->a == NULL)
	{
		perror("malloc fail!");
		exit(1);//malloc失败直接退出
	}
	ps->top = 0;//top指向栈顶元素的下一个位置
	ps->capacity = 4;
}

2.4栈的销毁

由于栈是动态开辟出来的空间，所以当栈使用完成之后，需要把栈销毁。

void STDestroy(ST* ps)
{
	assert(ps);

	free(ps->a);
	ps->a = NULL;
	ps->top = 0;
	ps->capacity = 0;
}

2.5入栈

void STPush(ST* ps, STDataType x)
{
	assert(ps);

    //若栈满了，则需要扩容
	if (ps->top == ps->capacity)
	{
        //2倍扩容
		STDataType* tmp = (STDataType*)realloc(ps->a, 2 * sizeof(STDataType) * ps->capacity);
		if (tmp == NULL)
		{
			perror("realloc fail!");
			exit(1);
		}
		ps->a = tmp;
		ps->capacity = 2 * ps->capacity;
	}
    //插入数据
	ps->a[ps->top] = x;
    //top往后走
	ps->top++;
}

2.6出栈

void STPop(ST* ps)
{
	assert(ps);
	assert(!STEmpty(ps));//栈为空时不能出栈

	ps->top--;
}

 

2.7获取栈的有效数据个数

top下标刚好就是栈的有效数据个数，所以直接返回top就可以了。

int STSize(ST* ps)
{
	assert(ps);
	return ps->top;
}

 2.8判断栈是否为空

若栈为空，则返回非0结果，若栈不为空，则返回0。

bool STEmpty(ST* ps)
{
	assert(ps);
	return ps->top == 0;
}

2.9访问栈顶元素

由于top指向栈顶元素的下一个位置，所以返回栈顶元素需要访问top-1位置的元素。

STDataType STTop(ST* ps)
{
	assert(ps);
	assert(!STEmpty(ps));//栈为空不能访问
	return ps->a[ps->top - 1];
}

 2.10栈的测试用例

需要注意的是，由于栈后进先出的性质，我们不能直接遍历栈，只能先访问栈顶元素，再出栈，才能访问下一个元素。

void TestStack()
{
	ST st;
	STInit(&st);

	STPush(&st, 1);
	STPush(&st, 2);
	STPush(&st, 3);
	STPush(&st, 4);
	STPush(&st, 5);
	while (!STEmpty(&st))
	{
		printf("%d ", STTop(&st));
		STPop(&st);
	}

	STDestroy(&st);
}

int main()
{
	TestStack();

	return 0;
}

以上是关于栈的一些知识，如有不对，欢迎指正！