《数据结构》链栈的表示和实现

/*
 链栈是运算受限的单链表，只能在链表 '头部' 操作
 - 基础知识：
 - 链表的头指针就是栈顶
 - 不需要头结点
 - 基本不存在栈满的情况
 - 空栈相当于头指针指向空
 - 插入和删除仅在栈顶处执行
*/

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

// 函数结果状态代码
#define TRUE 1
#define FALSE 0
#define OK 1
#define ERROR 0
#define INFEASIBLE -1

typedef int Status;    // Status 是函数的类型,其值是函数结果状态代码，如 OK 等
typedef int Boolean;   // Boolean 是布尔类型,其值是 TRUE 或 FALSE
typedef int SElemType; // 定义栈元素类型为整型

typedef struct StackNode
{
    SElemType data;         /* 数据域 */
    struct StackNode *next; /* 指针域 */
} StackNode, *LinkStack;

// 链栈的基本操作(9个)
#define DestroyStack ClearStack // DestroyStack()ClearStack()的操作是一样的

/* 链栈的初始化 */
Status InitStack(LinkStack *S)
{ // 构造一个空栈 S ，栈顶指针为空
    S = NULL;

    return OK;
}

/* 判断链栈是否为空 */
Status StackEmpty(LinkStack S)
{ // 若栈 S 为空栈，则返回 TRUE，否则返回 FALSE
    if (S)
        return FALSE;
    else
        return TRUE;
}

void ClearStack(LinkStack *S)
{ // 把S置为空栈
    LinkStack p;
    while (*S)
    {
        p = *S;
        *S = (*S)->next;
        free(p);
    }
}

int StackLength(LinkStack S)
{ // 返回 S 的元素个数，即栈的长度
    int i = 0;
    LinkStack p = S;
    while (p)
    {
        p = p->next;
        i++;
    }
    return i;
}

Status GetTop(LinkStack S, SElemType *e)
{ // 若栈不空，则用 e 返回 S 的栈顶元素，并返回 OK；否则返回 ERROR
    if (S == NULL)
        return ERROR;

    *e = S->data;
    return OK;
}

/* 链栈的入栈 */
Status Push(LinkStack *S, SElemType e)
{ // 插入元素 e 为新的栈顶元素
    LinkStack p;
    p = (LinkStack)malloc(sizeof(StackNode)); /* 为新的结点分配空间,生成结点*/
    if (p == NULL)
    {
        printf("malloc failed\n");
        return ERROR;
    }

    p->data = e;
    p->next = *S;
    *S = p;

    return OK;
}

/* 链栈的出栈 */
Status Pop(LinkStack *S, SElemType *e)
{ // 若栈不空，则删除 S 的栈顶元素，用 e 返回其值，并返回 OK；否则返回 ERROR
    if ((*S) == NULL)
        return ERROR;

    LinkStack p;

    *e = (*S)->data;
    p = *S;
    *S = (*S)->next;
    free(p);

    return OK;
}

void print(SElemType c)
{
    printf("%d ", c);
}

void main(int argc, char *argv[])
{
    int j;
    LinkStack s;
    SElemType e;
    InitStack(&s);           // 初始化栈 s
    for (j = 1; j <= 5; j++) // 将 2,4,6,8,10 入栈
        Push(&s, 2 * j);
    Pop(&s, &e);
    printf("pop stack is %d\n", e);
    GetTop(s, &e);
    printf("top value is %d\n", e);

    Pop(&s, &e);
    printf("pop stack is %d\n", e);
    GetTop(s, &e);
    printf("top value is %d\n", e);
}

/*
编译：gcc test.c -o test
执行:./test
结果：
pop stack is 10
top value is 8
pop stack is 8
top value is 6
*/

/********************** end of file ********************************/