线索二叉树(数据结构)

线索二叉树

#include <stdlib.h>
#include <iostream>
#include <stdio.h>

//线索二叉树结点
typedef struct ThreadNode{
	int data;
	struct ThreadNode *lchild,*rchild;
	int ltag,rtag;//左右线索的标志 
}ThreadNode,*ThreadTree;

//初始化一棵树 
void Create(ThreadTree &T)
{
	int d;
	printf("输入结点(按0为空结点):");
	scanf("%d",&d);
	if(d!=0)
	{
		T = (ThreadTree)malloc(sizeof(ThreadNode));
		T->data=d;
		T->ltag=0;
		T->rtag=0;
		Create(T->lchild);
		Create(T->rchild);
	}
	else{
		T=NULL;
		return;
	}
}

//线索化 
ThreadNode *pre;//定义全局变量pre,指向当前访问节点的前驱 

void visit(ThreadNode *q){
	if(q->lchild==NULL){//左子树为空，建立前驱线索 
		q->lchild=pre;
		q->ltag=1;
	}
	if(pre!=NULL&&q->rchild==NULL){
		pre->rchild=q;//建立前驱节点的后继线索 
		pre->rtag=1;
	}
	pre=q;
}

//中序遍历二叉树，一边遍历树一边线索化
void InThread(ThreadTree &T){
	if(T!=NULL){
		InThread(T->lchild);//中序遍历左子树 
		visit(T);//访问根节点 
		InThread(T->rchild);//中序遍历右子树 
	}
} 

//开始创建线索二叉树 
void CreateInTread(ThreadTree &T){
	pre=NULL;//pre初始值设置为NULL 
	if(T!=NULL){//非空二叉树才能线索化 
		InThread(T);//中序线索化二叉树
		if(pre->rchild==NULL){
			pre->rtag=1;//处理遍历的最后一个结点 
		} 
	}
}

//找到以P为根的子树中，第一个被中序遍历的结点
ThreadNode *FirstNode(ThreadNode *P){
	//循环找到最左下结点(不一定是叶结点)
	while(P->ltag=0) P=P->lchild;
	return P; 
} 

//在线索二叉树中找到结点P的后继结点
ThreadNode *NextNode(ThreadNode *P){
	//右子树中最左下的点
	if(P->rtag==0) return FirstNode(P->rchild);
	else return P->rchild;//rtage==1，直接返回后继结点 
}

//对中序线索二叉树进行中序遍历(利用线索实现，非递归)
void InOrder(ThreadNode *T){
	for(ThreadNode *P=FirstNode(T);P!=NULL;P=NextNode(P))
		printf("%d",P->data);//访问P结点 
}

int main(){
	ThreadTree T;
	Create(T);
	CreateInTread(T);
	InOrder(T);
	return 0;
}