bo6-3.cpp

清华大学《数据结构》教材第二版对应的C++教学程序

 // bo6-3.cpp 二叉树的二叉线索存储(存储结构由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
   if(!(Thrt=(BiThrTree)malloc(sizeof(BiThrNode))))
     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;
 }