数据结构 ：中序线索二叉树的遍历、找前驱后继

#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;
}