【代码 C/C++】线索二叉树(前序、中序、后序线索化)

正式开始打代码前的一些小知识

线索二叉树

线索二叉树使用值为NULL的左孩子指针或者右孩子指针来存储该结点的前驱和后继。
这里前驱和后继是在前序、中序或者后序遍历结果下的前驱和后继。
其中，左孩子指针用来存储前驱，右孩子指针用来存储后继，用两个标记：ltag和rtag（值为0，表示指针指向孩子结点，值为1，表示指针指向结点的前驱或后继）。
前序、中序、后序线索化与前序、中序、后序遍历大致类似，只要注意前序线索化有可能造成无线循环，需要多设定一些条件。

关于代码实现和应用场景

层次建树建立的都是完全二叉树，个人认为这里使用普通的二叉树更好一些，因此本文使用自己随便画的一个7结点二叉树进行线索化。

线索二叉树在工业上没有任何用处。

代码部分

代码是和王道书上差不多的

结构体

typedef char ThreadElemType;

typedef struct ThreadNode {
	ThreadElemType data;
	ThreadNode* lchild, * rchild;
	int ltag, rtag;
}ThreadNode, * ThreadTree;

全局变量

//建立一个7个结点的二叉树，结点由arr数组存储
ThreadNode* arr[7];
//全局变量pre，辅助指针，用于线索化时记录前一个结点
ThreadNode* pre = NULL;

建立一个7结点二叉树

//建立一个7个结点的二叉树
void BuildBiTree()
{	
	for (int i = 0; i < 7; i++)
	{
		arr[i] = (ThreadNode*)calloc(1, sizeof(ThreadNode));
		arr[i]->data = 'a' + i;
	}
	arr[0]->lchild = arr[1];
	arr[0]->rchild = arr[2];
	arr[1]->lchild = arr[3];
	arr[1]->rchild = arr[4];
	arr[2]->rchild = arr[5];
	arr[5]->lchild = arr[6];
}

visit()函数：用于对输入的单个结点线索化

//访问结点并将其空着的指针填上前驱或后继
void visit(ThreadNode* &p)
{

	if (p->lchild == NULL)
	{
		p->lchild = pre;
		p->ltag = 1;
	}
	if (pre != NULL && pre->rchild == NULL)
	{
		pre->rchild = p;
		pre->rtag = 1;
	}
	pre = p;
	//printf("%c\n", pre->data);
}

前序线索化

//前序线索化
void PreThread(ThreadTree T)
{
	if (T)
	{
		visit(T);
		if (T->ltag == 0)
		{
			PreThread(T->lchild);
		}
		if (T->rtag == 0)
		{
			PreThread(T->rchild);
		}
	}
}

//建立前序二叉树
void CreatePreThread(ThreadTree T)
{
	pre = NULL;
	if (T)
	{
		PreThread(T);
		if (pre->rchild == NULL)
		{
			pre->rtag = 1;
		}
	}
	
}

中序线索化

//中序线索化
void InThread(ThreadTree T)
{
	if (T)
	{
		InThread(T->lchild);
		visit(T);
		InThread(T->rchild);
	}
}

//建立中序线索二叉树
void CreateInThread(ThreadTree T)
{
	pre = NULL;
	if (T != NULL)
	{
		InThread(T);
		//处理遍历的最后一个结点的后继，将rtag设为1
		if (pre->rchild == NULL)
		{
			pre->rtag = 1;
		}
	}
}

后序线索化

//后序线索化
void PostThread(ThreadTree T)
{
	if (T)
	{

		PostThread(T->lchild);
		PostThread(T->rchild);
		visit(T);
	}
}

//建立后序线索二叉树
void CreatePostThread(ThreadTree T)
{
	pre = NULL;
	if (T != NULL)
	{
		PostThread(T);
		//处理遍历的最后一个结点的后继，将rtag设为1
		if (pre->rchild == NULL)
		{
			pre->rtag = 1;
		}
	}
}

打代码过程遇到的困难

关于前序线索二叉树的建立问题（无限循环）

最开始写的代码是照着中序线索二叉树稍微改了一下，具体如下

//前序线索化
void PreThread(ThreadTree T)
{
	if (T)
	{
		visit(T);
		PreThread(T->lchild);
		PreThread(T->rchild);
	}
}

这个代码有这样一个问题：
如下图，当d的左孩子指针通过visit(d)函数指向b以后，此时pre=d;

下一个要运行的函数是PreThread(d->lchild)，原本应该PreThread(NULL)，现在变成了PreThread(b)，这个函数运行visit(b)函数后，pre指向b，并且继续运行PreThread(b->lchild)，即PreThread(d)，visit(d)函数检测到d左孩子不是NULL，而右孩子为空，visit(d)函数判断后将d的右孩子指针继续指向b，然后一直循环下去。

因此需要使用ltag和rtag来避免这一情况。