c++,二叉树

#include <iostream>
#include <algorithm>
#include <string>
#include <cmath>
#include <time.h>

using namespace std;

typedef struct Node {
    int key,ltag,rtag;
    Node *lchild,*rchild;
} Node;

Node *getNewNode(int key) {
    Node *p=(Node*)malloc(sizeof(Node));
    p->key=key;
    p->ltag=p->rtag=0;
    p->lchild=p->rchild=NULL;
    return p;
}

Node *insert(Node *root,int key) {
    if(!root)return getNewNode(key);
    if(rand()%2)root->lchild=insert(root->lchild,key);
    else root->rchild=insert(root->rchild,key);
    return root;
}

void clear(Node *root) {
    if(!root)return ;
    clear(root->lchild);
    clear(root->rchild);
    free(root);
    return ;
}

void pre_order(Node *root) {
    if(!root)return ;
    cout<<root->key<<" ";
    if(!root->ltag)pre_order(root->lchild);
    if(!root->rtag)pre_order(root->rchild);
    return ;
}

void in_order(Node *root) {
    if(!root)return ;
    if(!root->ltag)in_order(root->lchild);
    cout<<root->key<<" ";
    if(!root->rtag)in_order(root->rchild);
    return ;
}

void post_order(Node *root) {
    if(!root)return ;
    if(!root->ltag)post_order(root->lchild);
    if(!root->rtag)post_order(root->rchild);
    cout<<root->key<<" ";
    return ;
}

Node *pre_node=NULL,*inorder_root=NULL;

void __build_inorder_thread(Node *root) {
    if(!root)return ;
    if(!root->ltag)__build_inorder_thread(root->lchild);
    if(inorder_root==NULL)inorder_root=root;
    if(root->lchild==NULL) {
        root->lchild=pre_node;
        root->ltag=1;
    }
    if(pre_node&&pre_node->rchild==NULL) {
        pre_node->rchild=root;
        pre_node->rtag=1;
    }
    pre_node=root;
    if(!root->rtag)__build_inorder_thread(root->rchild);
    return ;
}

void build_inorder_thread(Node *root) {
    __build_inorder_thread(root);
    pre_node->rchild=NULL;
    pre_node->rtag=1;
    return ;
}

Node *getNext(Node *root){
    if(root->rtag==1)return root->rchild;
    root=root->rchild;
    while(root->ltag==0&&root->lchild)root=root->lchild;
    return root;
}

int main() {
    srand(time(0));
    Node *root=NULL;
    for(int i=0; i<10; i++) {
        root=insert(root,rand()%100);
    }
    pre_node=inorder_root=NULL;
    build_inorder_thread(root);
    pre_order(root);
    printf("\n");
    in_order(root);
    printf("\n");
    post_order(root);
    cout<<endl;
    Node *node=inorder_root;
    while(node) {
        cout<<node->key<<" ";
        node=getNext(node);
    }
    clear(root);
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

慕容晓开

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值