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

typedef struct Tree
{
    int data,Ltag,Rtag;//数据，左右标志 
    struct Tree *LChild,*RChild;//左右孩子指针域 
}Tree,*Trees;
Trees pre=NULL;

void InitTree(Trees *boot)//初始化二叉树，只有一个根节点，无左右孩子 
{
    *boot=(Trees)malloc(sizeof(Tree));
    (*boot)->LChild=(*boot)->RChild=NULL;
    (*boot)->Ltag=(*boot)->Rtag=0; 
}

void CreateTree(Trees &boot)//创建二叉树，具体操作参考上一篇《树与二叉树》 
{
    int ch;
    scanf("%d",&ch);
    if(ch==0)
        boot=NULL;
    else
    {
        boot=(Trees)malloc(sizeof(Tree));
        boot->data=ch;
        boot->LChild=boot->RChild=NULL;
        boot->Ltag=boot->Rtag=0;
        CreateTree(boot->LChild);
        CreateTree(boot->RChild);
    }
}
//添加线索 
void InThread(Trees &boot)
{
    if(boot!=NULL)
    {
        InThread(boot->LChild);//线索化左子树 
        if(boot->LChild==NULL)
        {
            boot->Ltag=1;
            boot->LChild=pre;//置前驱线索 
        }
        if(pre!=NULL&&pre->RChild==NULL)
        {
            pre->RChild=boot;
            pre->Rtag=1;
        }
        pre=boot;//当前访问节点为下一个访问节点的前驱 
        InThread(boot->RChild);//线索化右子树 
    }
} 

Trees InOrderThread(Trees &rt)//建头结点 
{
    Trees thrt;
    if(!(thrt=(Trees)malloc(sizeof(Tree))))
    {
        printf("头结点创建失败.\n");
        exit(1);
    }
    
    thrt->Ltag=0;//等于零指向左孩子  1
    thrt->Rtag=1;//等于一指向遍历的前驱 2
    thrt->RChild=thrt;//右回指针,指向自己(头结点) 3
    if(!rt)
        thrt->LChild=thrt;//若二叉树为空，建立左回指针(也指向头结点)
    else
    {
        thrt->LChild=rt;//rt==boot，也就是二叉树的根 
        pre=thrt;//pre指向头结点 
        InThread(rt);//为二叉树加线索 
        pre->RChild=thrt;//同上1
        pre->Rtag=1;//同上2
        thrt->RChild=pre;//同上3
    }
    return thrt;
}

//中序找前驱 
void InPre(Trees boot)
{
    Trees q=NULL;
    if(boot->Ltag==1)
        pre=boot->LChild;
    else
    {
        for(q=boot->LChild;q->Rtag==0;q=q->RChild)
            pre=q;    
    }    
    if(pre)
        printf("中序找到的前驱为：%d\n",pre->data);    
    else
        printf("无前驱.\n");
}

//中序找后继 
void InNext(Trees boot)
{
    Trees q=NULL;
    if(boot->Rtag==1)
        pre=boot->RChild;
    else
    {
        for(q=boot->RChild;q->Ltag==0;q=q->LChild)
            pre=q;
    }
    if(pre)
        printf("中序找到的后继为：%d\n",pre->data);
    else
        printf("中序遍历无后继.\n");
}

//中序遍历序线索树上的第一个结点 
Trees InFirst(Trees boot)
{
    Trees p=boot;
    if(!p)
        return NULL;
    while(p->Ltag==0)
        p=p->LChild;
    return p;
}

//中序遍历线索二叉树 
void TInOrder(Trees &thrt)
{
    Trees p;
    p=thrt->LChild;
    while(p!=thrt)
    {
        while(p->Ltag==0)
            p=p->LChild;
        printf("%d ",p->data);
        while(p->Rtag==1&&p->RChild!=thrt)
        {
            p=p->RChild;
            printf("%d ",p->data);        
        }
        p=p->RChild;
    }
    printf("\n");
}

int main()
{
    Trees boot=NULL;
    printf("创建二叉树，输入零结束:\n");
    CreateTree(boot);
    
    Trees thrt;//头结点 
    thrt=InOrderThread(boot);
    TInOrder(thrt);//中序遍历线索二叉树 
    
    InPre(boot);//中序找前驱 
    
    InNext(boot);//中序找后继 
    
    Trees bt=InFirst(boot);//中序遍历序线索树上的第一个结点
    printf("中序遍历序线索树上的第一个结点为:%d\n",bt->data);
    
    return 0;
}