建立中序线索二叉树并遍历

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

#define OVERFLOW -2

typedef struct TBTNode
{
    char data;
    int ltag, rtag;                  // 线索标记
    struct TBTNode* lchild, *rchild; // 孩子或线索指针
} TBTNode;


void
Create_TBT(TBTNode* & T)
{
    /* 按先序输入二叉线索树中结点的值,构造二叉线索树T */
    /* "#"表示空结点 */
    char ch;
    scanf("%c", &ch);

    if(ch == '#')
        T = NULL;
    else
    {
        T = (TBTNode*)malloc(sizeof(TBTNode));
        if(!T)
            exit(OVERFLOW);
        T->data = ch;  /* 生成根结点(先序) */
        Create_TBT(T->lchild);/* 递归构造左子树 */
        if(T->lchild) /* 有左孩子 */
            T->ltag = 0;
        Create_TBT(T->rchild); /* 递归构造右子树 */
        if(T->rchild) /* 有右孩子 */
            T->rtag = 0;
    }
}

TBTNode* pre;   // 全局变量,始终指向刚刚访问过的结点
void
Thread(TBTNode*& p) // 中序遍历进行中序线索化
{
    if (p != NULL)
    {
        Thread(p->lchild);  // 左子树线索化
        if (p->lchild == NULL)  // 前驱线索
        {
            p->lchild = pre;    // 建立当前结点的前驱线索
            p->ltag = 1;
        }
        else
            p->ltag = 0;
        if (pre->rchild == NULL)  // 后继线索
        {
            pre->rchild = p;         // 建立前驱结点的后继线索
            pre->rtag = 1;
        }
        else
            pre->rtag = 0;
        pre = p;             // 保持pre指向p的前驱
        Thread(p->rchild);   // 右子树线索化
    }
}

TBTNode*
Create_Thread(TBTNode* & b)
{
    /* 中序遍历二叉树b,并将其中序线索化,root指向头结点 */
    TBTNode* root;
    root = (TBTNode*)malloc(sizeof(TBTNode));  // 创建头结点
    if(!root)
        exit(OVERFLOW);
    root->ltag = 0;
    root->rtag = 1;
    root->rchild = root;        /* 右指针回指 */
    if (b == NULL)                           /* 若二叉树空，则左指针回指 */
        root->lchild = root;
    else
    {
        root->lchild = b;
        pre = root;                     // *pre是*p的前驱结点，供加线索用

        Thread(b);                     /* 中序遍历进行中序线索化 */

        pre->rchild = root;              // 处理最后一个结点
        pre->rtag = 1;
        root->rchild = pre;             // 头结点右线索化
    }
    return root;
}

void
ThInOrder(TBTNode* tb)
{
    /* 中序遍历二叉线索树tb(头结点)的非递归算法 */
    TBTNode* p = tb->lchild;  // p指向根结点
    while (p != tb)      /* 空树或遍历结束时,p==tb */
    {
        while (p->ltag == 0)            // 找开始结点
            p = p->lchild;
        printf("%c ", p->data);           // 访问开始结点

        while (p->rtag == 1 && p->rchild != tb)
        {
            p = p->rchild;
            printf("%c ", p->data);
        }
        p = p->rchild;
    }
}

int
main()
{
    TBTNode *root, *b;
    printf("请将二叉树中的结点按从上到下从左到右的次序输入:\n");
    Create_TBT(b);              /* 按先序产生二叉树 */
    root = Create_Thread(b);        /* 中序遍历，并中序线索化二叉树 */
    printf("中序遍历二叉线索树的结果为:\n");
    ThInOrder(root);                    /* 中序遍历二叉线索树 */
    return 0;
}