先总结一下,当通过先序遍历线索化二叉树的时候,我们在递归遍历左子树以及右子树的时候都需要加以判断,否则就会出现死循环------爱滴魔力转圈圈!
首先我们先来看一下经常写的先序二叉树线索化的代码(我是看着某站上面王道的视频写的):
#include<iostream>
using namespace std;
//线索链表
//线索二叉树结点
typedef struct ThreadNode
{
char data;
ThreadNode* lchild, * rchild;
//ltag 左线索标志, rtag又线索标志
//tag == 0,表示child指针指向孩子
//tag == 1,表示child指针指向线索
//lchild 充当线索指向的就是前驱,rchild 充当线索指向的就是后继
int ltag, rtag;
}*ThreadTree;
//全局变量pre, 指向当前访问的结点的前驱
ThreadNode* pre = NULL;
void visit(ThreadNode* q)
{
if (q->lchild == NULL)
{
q->lchild = pre; //建立当前结点的前驱线索
q->ltag = 1;
}
if (pre != NULL && pre->rchild == NULL)
{
pre->rchild = q; //建立前驱结点的后继线索
pre->rtag = 1;
}
pre = q;
}
//先序遍历 根左右 (先根遍历)
void PreThread(ThreadTree t)
{
if (t != NULL)
{
visit(t); //访问根节点
if (t->ltag == 0) //判断当前结点的lchild不是前驱线索
{
PreThread(t->lchild); //递归遍历左子树
}
PreThread(t->rchild); //递归遍历右子树
}
}
//先序线索化二叉树
void CreatePreThread(ThreadTree t)
{
//pre = NULL;//pre的初始值为空
if (t != NULL) //非空二叉树进行线索化
{
PreThread(t); //先序线索化二叉树
if (pre->rchild == NULL)
{
pre->rtag = 1; //处理遍历的最后一个结点
}
}
}
上面的代码是网上常看到的先序线索化二叉树,看起来确实没啥问题,但是自己加点数据对线索二叉树进行遍历的时候一运行发现出错了,自己想了半天,然后调试着看了很久,找到一个容易出错的地方,先看下图
//先序遍历 根左右 (先根遍历)
void PreThread(ThreadTree t)
{
if (t != NULL)
{
visit(t); //访问根节点
if (t->ltag == 0) //判断当前结点的lchild不是前驱线索
{
PreThread(t->lchild); //递归遍历左子树
}
PreThread(t->rchild); //递归遍历右子树
}
}
先序二叉树遍历的时候是根左右,在遍历过程中会先对根结点,以及根结点的左子树进行线索化,但是看下面的一种情况(图片来自王道数据结构考研视频):
我们对结点“D”进行访问的时候,即调用visit()函数的时候,此时pre指针指向的是结点“B”,通过判断结点D的左孩子为空,因此建立左索引,将其前驱指向结点B,即D->lchild = B,但接下来执行PreThread(D->lchild) 的时候如果不加判断,此时已经将左指针指向B,那么就会在此处进行不断循环,所以我们在递归遍历左子树的时候需要加以判断其前结点的lchild不是前驱线索,其实当递归遍历右结点的时候同时也需要进行遍历,否则也会出现转圈的问题,例如当访问结点F的时候,visit(F),其左右孩子结点都为空,因此有:F->lchild = C; F->ltag = 1; 以及C->rchild = F; C->rtag = 1;,遍历完F结点,接着就会运行PreThread(C->rchild);,如果此时不加以判断,就会同上面所说的一样出现“爱滴魔力转圈圈”问题,找到问题,接下来把代码完整附上:
#include<iostream>
using namespace std;
//线索链表
//线索二叉树结点
typedef struct ThreadNode
{
char data;
ThreadNode* lchild, * rchild;
//ltag 左线索标志, rtag又线索标志
//tag == 0,表示child指针指向孩子
//tag == 1,表示child指针指向线索
//lchild 充当线索指向的就是前驱,rchild 充当线索指向的就是后继
int ltag, rtag;
}*ThreadTree;
//全局变量pre, 指向当前访问的结点的前驱
ThreadNode* pre = NULL;
void visit(ThreadNode* q)
{
if (q->lchild == NULL)
{
q->lchild = pre; //建立当前结点的前驱线索
q->ltag = 1;
}
if (pre != NULL && pre->rchild == NULL)
{
pre->rchild = q; //建立前驱结点的后继线索
pre->rtag = 1;
}
pre = q;
}
//先序遍历 根左右 (先根遍历)
void PreThread(ThreadTree t)
{
if (t != NULL)
{
visit(t); //访问根节点
if (t->ltag == 0) //判断当前结点的lchild不是前驱线索
{
PreThread(t->lchild); //递归遍历左子树
}
if (t->rtag == 0) //判断当前结点的rchild不是后继线索
{
PreThread(t->rchild); //递归遍历右子树
}
}
}
//先序线索化二叉树
void CreatePreThread(ThreadTree t)
{
//pre = NULL;//pre的初始值为空
if (t != NULL) //非空二叉树进行线索化
{
PreThread(t); //先序线索化二叉树
if (pre->rchild == NULL)
{
pre->rtag = 1; //处理遍历的最后一个结点
}
}
}
通过验证,完美通过!运行结果是第三张图先序线索二叉树的遍历结果