链式堆栈

初学数据结构，一个小程序（包括修改过程）

LinStack.h

/*定义结点*/
typedef struct ListNode      
{
	DataType data;                //定义数据域
	struct ListNode *next;        //定义指针域
}ListNode;

/*定义头结点*/
ListNode *Head()
{
	ListNode *p = (ListNode*)malloc(sizeof(ListNode));
	if(p == NULL)
	{
		printf("内存申请失败退出！");
		exit(0);
	}

	p->next = NULL;

	return p;
}

/*定义普通结点*/
ListNode *Node(DataType data)
{
	ListNode *p = (ListNode*)malloc(sizeof(ListNode));
	if(p == NULL)
	{
		printf("内存申请失败退出！");
		exit(0);
	}

	p->data = data;
	p->next = NULL;

	return p;
}

/*初始化堆栈*/
ListNode *Initiate()
{
	ListNode *head = Head();

	return head;
}

/*入栈*/
ListNode *Push(DataType data,ListNode *head)
{
	ListNode *p = Node(data);    //申请一个结点，保存待入栈的数据

	p->next = head->next;
	head->next = p;

	return p;
}

/*出栈*/
DataType Pop(ListNode *head)           /*这个和上一个不知道系统是怎么知道是头结点的*/   //应该是在 main.c 中传值的吧
{
	if(head->next == NULL)
	{
		printf("堆栈已空，无法出栈！");
		return 0;
	}
	else
	{
		ListNode *p = NULL;
		p = head->next;
		head->next = head->next->next;

		free(p);
		//p = NULL;

		return p->data;          /*不知道这样对不对，释放后用（书上是这样用的）*/
	}
}

/*取栈顶元素*/
DataType GetTop(ListNode *head)
{
	if(head->next == NULL)
	{
		printf("堆栈为空,无元素可取！");
		return 0;
	}
	else
	{
		return head->next->data;
	}
}

/*判断堆栈非空否*/
int NotEmpty(ListNode *head)
{
	if(head->next == NULL)
	{
		//printf("堆栈为空！\n");
		return 0;                    //为空时返回 0
	}
	else
	{
		//printf("堆栈非空！");
		return 1;                    //非空是返回 1
	}
}

main.c

#include <stdio.h>
#include <stdlib.h>

typedef int DataType;
#include "LinStack.h"

void main()
{
	DataType data;
	ListNode *myStack = Initiate();

	while(1)
	{
		printf("请输入要入栈的数据元素（输入 0 时结束输入）：");
		scanf("%d",&data);
		if(data == 0)
		{
			break;
		}

		Push(data,myStack);
	}

	if(NotEmpty(myStack)== 0)
	{
		printf("堆栈为空！\n");
	}
	while(NotEmpty(myStack) == 1)
	{
		printf(" %d ",GetTop(myStack));         //必需要 printf()
		Pop(myStack);
	}
}