数据结构学习【线索二叉树C++ 先序&中序遍历(不使用栈)】

数据结构学习【线索二叉树C++ 先序&中序遍历(不使用栈)】

本次实现的是线索二叉树。插入方式采用的是二叉排序树的插入方式。实现了二叉树的先序&中序遍历(采用线索来遍历,而不是借助栈)。

之所以不实现后序遍历,是因为后序线索二叉树对于rtag=0的结点(即右指针指向的是右孩子的结点)难以找到其后继结点,除非使用三叉链表。故此处只实现了后序线索二叉树的建立,但是未实现后序遍历。

#include 
using namespace std;

typedef struct ThreadNode
{
    int data;
    struct ThreadNode *lchild, *rchild;
    int ltag,rtag;
} ThreadNode,*ThreadTree;

bool Insert(ThreadTree &root, int data);  // 根据二叉排序树插入
//先序构造线索二叉树
void CreatPreThread(ThreadTree root);  
void PreThread(ThreadTree root,ThreadNode* &pre);
//中序构造线索二叉树
void CreatInThread(ThreadTree root);  
void InThread(ThreadTree root,ThreadNode* &pre);
//后序构造线索二叉树
void CreatPostThread(ThreadTree root);  
void PostThread(ThreadTree root,ThreadNode* &pre);


void Traversal(ThreadNode *root, int type); // 0: 先序, 1: 中序, (采用线索二叉树的方式遍历,无需栈)

void PreorderTraversal(ThreadNode *root);
ThreadNode* NextPreNode(ThreadNode* p);

void InorderTraversal(ThreadNode *root);
ThreadNode* NextInNode(ThreadNode* p);

// Code

int main()
{
    ThreadTree root = nullptr;
    Insert(root, 6);
    Insert(root, 4);
    Insert(root, 5);
    Insert(root, 8);
    Insert(root, 7);
    Insert(root, 9);
    
    printf("二叉排序树");

    CreatPreThread(root);
    printf("\n先序:");
    Traversal(root,0);

    // CreatInThread(root);
    // printf("\n中序:");
    // Traversal(root,1);

    //CreatPostThread(root);

    system("pause");
}
// 根据二叉排序树插入
bool Insert(ThreadTree &p, int data)
{
    if (p == nullptr)
    {
        p = (ThreadNode *)malloc(sizeof(ThreadNode));
        p->data = data;
        p->lchild = nullptr;
        p->rchild = nullptr;
        p->ltag=0;
        p->rtag=0;
        return true;
    }
    if (data < p->data)
    {
        return Insert(p->lchild, data);
    }
    else if (data > p->data)
    {
        return Insert(p->rchild, data);
    }
    else
    {
        return false;
    }
}
//先序构造线索二叉树
void CreatPreThread(ThreadTree root){
    ThreadNode* pre=nullptr;
    if(root!=nullptr){
        PreThread(root,pre);
        if(pre->rchild==nullptr)
            pre->rtag=1;
    }
} 
void PreThread(ThreadTree root,ThreadNode* &pre){
    if(root!=nullptr){
        // 建立前驱线索
        if(root->lchild==nullptr){
            root->ltag=1;
            root->lchild=pre;
        }
        // 建立后继线索
        if(pre!=nullptr&&pre->rchild==nullptr){
            pre->rtag=1;
            pre->rchild=root;
        }
        pre=root;
        if(root->ltag==0)//避免循环
            PreThread(root->lchild,pre);
        PreThread(root->rchild,pre);
    }
}
//中序构造线索二叉树
void CreatInThread(ThreadTree root){
    ThreadNode* pre=nullptr;
    if(root!=nullptr){
        InThread(root,pre);
        if(pre->rchild==nullptr)
            pre->rtag=1;
    }
}
void InThread(ThreadTree root,ThreadNode* &pre){
    if(root!=nullptr){
        InThread(root->lchild,pre);
        // 建立前驱线索
        if(root->lchild==nullptr){
            root->ltag=1;
            root->lchild=pre;
        }
        // 建立后继线索
        if(pre!=nullptr&&pre->rchild==nullptr){
            pre->rtag=1;
            pre->rchild=root;
        }
        pre=root;
        InThread(root->rchild,pre);
    }
}
//后序构造线索二叉树
void CreatPostThread(ThreadTree root){
    ThreadNode* pre=nullptr;
    if(root!=nullptr){
        PostThread(root,pre);
        if(pre->rchild==nullptr)
            pre->rtag=1;
    }
}
void PostThread(ThreadTree root,ThreadNode* &pre){
    if(root!=nullptr){
        PostThread(root->lchild,pre);
        PostThread(root->rchild,pre);
        // 建立前驱线索
        if(root->lchild==nullptr){
            root->ltag=1;
            root->lchild=pre;
        }
        // 建立后继线索
        if(pre!=nullptr&&pre->rchild==nullptr){
            pre->rtag=1;
            pre->rchild=root;
        }
        pre=root;
    }
}

// 0: 先序, 1: 中序
void Traversal(ThreadNode *root, int type)
{
    switch (type)
    {
    case 0:
        PreorderTraversal(root);
        break;
    case 1:
        InorderTraversal(root);
        break;
    default:
        break;
    }
}
// 先序
void PreorderTraversal(ThreadNode *root){
   for(ThreadNode* p=root;p!=nullptr;p=NextPreNode(p)){
       visit(p);
   }
}
ThreadNode* NextPreNode(ThreadNode* p){
    if(p->rtag==1) return p->rchild;
    else{
        if(p->lchild!=nullptr&&p->ltag==0) return p->lchild;
        else return p->rchild;
    }
}
// 中序
void InorderTraversal(ThreadNode *root){
    // 找到第一个结点
    ThreadNode* tmp=root;
    while (tmp->ltag==0)
    {
        tmp=tmp->lchild;
    }
    // 遍历
    for(ThreadNode* p=tmp;p!=nullptr;p=NextInNode(p)){
       visit(p);
   }
}
ThreadNode* NextInNode(ThreadNode* p){
    if(p->rtag==1) return p->rchild;
    else{
        ThreadNode* tmp=p->rchild;
        while(tmp->ltag==0) tmp=tmp->lchild;
        return tmp;
    }
}


//访问结点
void visit(ThreadNode* p){
    printf("%d ",p->data);
}


本文插入数据形成的二叉树如图
数据结构学习【线索二叉树C++ 先序&中序遍历(不使用栈)】_第1张图片

你可能感兴趣的:(数据结构,数据结构,c++,排序算法,树结构)