#include <stdio.h>
#include <stdlib.h>
#define ElemType int
typedef struct ThreadNode{
ElemType data;
// 前驱线索->左孩子、后继线索->右孩子
struct ThreadNode *lchild, *rchild;
// tag为0表示孩子,tag为1表示线索
int ltag, rtag;
} ThreadNode, *ThreadTree;
ThreadNode *preNode = NULL; // 全局变量,用来记录当前结点的前驱结点
// 线索化处理每一个结点
void visit(ThreadNode *nowNode)
{
if(NULL == nowNode->lchild){ // 当前结点的前驱为空,为他建立前驱线索
nowNode->lchild = preNode;
nowNode->ltag = 1;
}
if(NULL != preNode && NULL == preNode->rchild){ // 为前驱结点建立后继线索
preNode->rchild = nowNode;
preNode->rtag = 1;
}
preNode = nowNode; // 后移,以便访问下一个结点
}
// 线索二叉树初始化
void InitThreadTree(ThreadTree &T)
{
T == NULL;
}
// 中序线索
void InThread(ThreadTree T)
{
if(NULL != T){
InThread(T->lchild); // 递归遍历左子树
visit(T); // 访问根节点
InThread(T->rchild); // 递归遍历右子树
}
}
// 建立二叉树结点并返回结点地址
ThreadNode* CreateNode(int data)
{
struct ThreadNode *node = (ThreadNode *)malloc(sizeof(ThreadNode));
node->ltag = 0;
node->rtag = 0;
node->lchild = NULL;
node->rchild = NULL;
node->data = data;
return node;
}
// 建立二叉树
void CreateBiTree(ThreadTree &T)
{
// 手动建立一个简单的二叉树
T = CreateNode(1);
T->lchild = CreateNode(2);
T->rchild = CreateNode(3);
T->lchild->lchild = CreateNode(4);
T->lchild->rchild = CreateNode(8);
T->rchild->lchild = CreateNode(5);
T->rchild->rchild = CreateNode(6);
}
// 建立中序线索二叉树
void CreateInThread(ThreadTree T)
{
preNode = NULL;
if(NULL != T){
InThread(T); // 这一行是哪种遍历方式就是哪种线索化方式
if(NULL == preNode->rchild){ // 遍历到的最后一个结点线索化
preNode->rtag = 1;
}
}
}
// 找到以p为根的子树中,第一个被中序遍历的结点
ThreadNode* FirstNode(ThreadNode *p)
{
while(0 == p->ltag){ // 循环找到最左下结点
p = p->lchild;
}
return p;
}
// 中序线索二叉树中找到结点p的后继结点
ThreadNode *NextNode(ThreadNode *p)
{
if(0 == p->rtag){
// 返回右子树中的最左下结点
return FirstNode(p->rchild);
}
else{
return p->rchild; // rtag == 1 直接返回后继线索
}
}
// 找到以p为根的子树中,最后一个被中序遍历的点
ThreadNode* LastNode(ThreadNode *p)
{
// 循环找到最右下结点(不一定是叶节点)
while(0 == p->rtag){
p = p->rchild;
}
return p;
}
// 在中序线索二叉树中找到p的前驱
ThreadNode* PreNode(ThreadNode *p)
{
if(p->ltag == 0){
// 返回左子树中最右下结点
return LastNode(p->lchild);
}
else{
return p->lchild;
}
}
// 打印结点信息
void PrintNode(ThreadNode *node)
{
if(NULL == node->lchild){
printf("结点前驱:NULL ");
}
else{
if(node->ltag == 1)
printf("结点前驱:%d ", node->lchild->data);
else{
ThreadNode *p1;
p1 = PreNode(node);
printf("结点前驱: %d ", p1->data);
}
}
printf("访问结点:%d ", node->data);
if(NULL == node->rchild){
printf("结点后继:NULL\n");
}
else{
if(node->rtag == 1)
printf("结点后继:%d\n", node->rchild->data);
else{
ThreadNode *p;
p = NextNode(node);
printf("结点后继: %d\n", p->data);
}
}
}
// 对中序线索二叉树进行中序遍历(非递归算法)
void InOrder(ThreadNode *T)
{
for(ThreadNode *p = FirstNode(T); p != NULL; p = NextNode(p)){
PrintNode(p);
}
}
// 对线索二叉树逆向中序遍历
void RevInOrder(ThreadNode *T)
{
for(ThreadNode *p = LastNode(T); p != NULL; p = PreNode(p)){
PrintNode(p);
}
}
int main()
{
ThreadTree T;
InitThreadTree(T);
CreateBiTree(T); // 建立二叉树
CreateInThread(T); // 建立线索二叉树
InOrder(T); // 中序遍历输出
printf("\n");
printf("反转逆序输出\n");
RevInOrder(T);
return 0;
}
数据结构 :中序线索二叉树的遍历、找前驱后继
最新推荐文章于 2023-09-15 23:34:45 发布