线索二叉树
简介
遍历树的以一定规则将二叉树中的结点排列成一个线性序列,得到树中结点的先序序列、中序序列或后序序列。这些线性序列中的每一个元素都有且仅有一个前驱结点和后继结点,注意第一个节点只有后继节点,最后一个节点只有前驱节点。
线索二叉树的结构
typedef struct ThreadNode {
ElemType data;
struct ThreadNode *lchild, *rchild; // 指向左右孩子节点
int ltag, rtag; // 左右线索标志 1 代表是线索节点,0代表是正常孩子节点
} ThreadNode, * ThreadTree;
ltag为0是指向该结点的左孩子,为1时指向该结点的前驱
rtag为0是指向该结点的右孩子,为1时指向该结点的后继
二叉树的中序线索化
如图将该二叉树线索化。
思路分析:
- 我们肯定需要中序遍历树中的每一个节点,判断节点是否需要线索化。(判断左右子树是否为NULL)
- 还需要建立一个指向当前节点前一个节点的指针pre,用来帮助前驱索引
- 如果左子树为空,需要建立前驱索引。右子树为空,需要建立后驱索引。
- 特别需要注意最后一个节点,需要单独处理一下
程序实例C语言实现
// 全局变量pre,指向当前访问节点的前驱
ThreadNode * pre = NULL;
// 构建这棵索引二叉树
void CreateInThread(ThreadTree tree) {
pre = NULL; // 保证多次建立线索时,pre为空
if (tree != NULL) {
InThread(tree); // 中序构建二叉树
if (pre->rchild == NULL) { // 最后一个节点需要额外处理
pre->rtag = 1; // 最后一个节点肯定为NULL,直接置为1
}
}
}
// 中序遍历二叉树
void InThread(ThreadTree tree) {
if(tree != NULL) {
InThread(tree->lchild);
visit(tree); // 和遍历类似,只不过这里变成了线索化的方法
InThread(tree->rchild);
}
}
// 对二叉树进行线索化,参数即为 判断是否需要 线索的节点
void visit(ThreadNode * node) {
if(node->lchild == NULL) { // 左子树为空,建立前驱所用
node->lchild = pre;
node->ltag = 1;
}
// 每次建立后继索引,肯定在前驱索引之后。需要用到下一个节点
if (pre != NULL && pre->rchild == NULL) {
pre->rchild = node; // 建立前驱节点的后继线索
pre->rtag = 1;
}
pre = node;
}
二叉树的先序线索化
类似于中序线索化只是将对节点的访问顺序变成了先序遍历。沿用上一个树。
// 先序遍历
void PreThread(ThreadTree tree) {
if(tree != NULL) {
visit(tree);
PreThread(tree->lchild);
PreThread(tree->rchild);
}
}
这里的代码会发生一个问题,运行时会死循环。原因在于,构建第D节点的前驱索引后,在遍历时会发生死循环。一直在B和D之间来回跳。因为将D的左孩子指向了B节点,这里遍历的代码需要修改。
// 先序遍历
void PreThread(ThreadTree tree) {
if(tree != NULL) {
visit(tree);
if (tree->ltag == 0) { // 增加一个判断,解决问题
PreThread(tree->lchild);
}
if (tree->rtag == 0) {
PreThread(tree->rchild);
}
}
}
二叉树的后序线索化
类似
void PostThread(ThreadTree tree) {
if(tree != NULL) {
PostThread(tree->lchild);
PostThread(tree->rchild);
visit(tree);
}
}
void CreatePostThread(ThreadTree tree) {
pre = NULL; // 保证多个线索时,pre为空
if (tree != NULL) {
PostThread(tree);
if (pre->rchild == NULL) { // 最后一个节点需要额外处理
pre->rtag = 1;
}
}
}