二叉树的二叉线索存储表示

转载：http://hi.baidu.com/fire_man/blog/item/0dd0dbcacf348884c81768a9.html

/* c6-3.h 二叉树的二叉线索存储表示 */

typedef enum{Link,Thread}PointerTag; /* Link(0):指针,Thread(1):线索 */

typedef struct BiThrNode

{

    TElemType data;

    struct BiThrNode *lchild,*rchild; /* 左右孩子指针 */

    PointerTag LTag,RTag; /* 左右标志 */

}BiThrNode,*BiThrTree;

/* bo6-3.c 二叉树的二叉线索存储(存储结构由c6-3.h定义)的基本操作 */

Status CreateBiThrTree(BiThrTree *T)

{ /* 按先序输入二叉线索树中结点的值,构造二叉线索树T */

    /* 0(整型)/空格(字符型)表示空结点 */

    TElemType h;

#if CHAR

    scanf("%c",&h);

#else

    scanf("%d",&h);

#endif

    if(h==Nil)

      *T=NULL;

    else

    {

      *T=(BiThrTree)malloc(sizeof(BiThrNode));

      if(!*T)

        exit(OVERFLOW);

      (*T)->data=h; /* 生成根结点(先序) */

      CreateBiThrTree(&(*T)->lchild); /* 递归构造左子树 */

      if((*T)->lchild) /* 有左孩子 */

        (*T)->LTag=Link;

      CreateBiThrTree(&(*T)->rchild); /* 递归构造右子树 */

      if((*T)->rchild) /* 有右孩子 */

        (*T)->RTag=Link;

    }

    return OK;

}

BiThrTree pre; /* 全局变量,始终指向刚刚访问过的结点 */

void InThreading(BiThrTree p)

{ /* 中序遍历进行中序线索化。算法6.7 */

    if(p)

    {

      InThreading(p->lchild); /* 递归左子树线索化 */

      if(!p->lchild) /* 没有左孩子 */

      {

        p->LTag=Thread; /* 前驱线索 */

        p->lchild=pre; /* 左孩子指针指向前驱 */

      }

      if(!pre->rchild) /* 前驱没有右孩子 */

      {

        pre->RTag=Thread; /* 后继线索 */

        pre->rchild=p; /* 前驱右孩子指针指向后继(当前结点p) */

      }

      pre=p; /* 保持pre指向p的前驱 */

      InThreading(p->rchild); /* 递归右子树线索化 */

    }

}

Status InOrderThreading(BiThrTree *Thrt,BiThrTree T)

{ /* 中序遍历二叉树T,并将其中序线索化,Thrt指向头结点。算法6.6 */

    *Thrt=(BiThrTree)malloc(sizeof(BiThrNode));

    if(!*Thrt)

      exit(OVERFLOW);

    (*Thrt)->LTag=Link; /* 建头结点 */

    (*Thrt)->RTag=Thread;

    (*Thrt)->rchild=*Thrt; /* 右指针回指 */

    if(!T) /* 若二叉树空，则左指针回指 */

      (*Thrt)->lchild=*Thrt;

    else

    {

      (*Thrt)->lchild=T;

      pre=*Thrt;

      InThreading(T); /* 中序遍历进行中序线索化 */

      pre->rchild=*Thrt;

      pre->RTag=Thread; /* 最后一个结点线索化 */

      (*Thrt)->rchild=pre;

    }

    return OK;

}

Status InOrderTraverse_Thr(BiThrTree T,Status(*Visit)(TElemType))

{ /* 中序遍历二叉线索树T(头结点)的非递归算法。算法6.5 */

    BiThrTree p;

    p=T->lchild; /* p指向根结点 */

    while(p!=T)

    { /* 空树或遍历结束时,p==T */

      while(p->LTag==Link)

        p=p->lchild;

      if(!Visit(p->data)) /* 访问其左子树为空的结点 */

        return ERROR;

      while(p->RTag==Thread&&p->rchild!=T)

      {

        p=p->rchild;

        Visit(p->data); /* 访问后继结点 */

      }

      p=p->rchild;

    }

    return OK;

}