C++中通过随机与递归创建二叉树并以特定格式打印

#include<iostream>
#include<stdlib.h> 
#include <time.h>
using namespace std;

// 定义元素类型
typedef int element;

// 定义二叉树节点结构体
typedef struct btree {
    struct btree* lchild; // 左子树指针
    struct btree* rchild; // 右子树指针
    element data; // 节点存储的数据
}*BTree, btree;

// 初始化二叉树
BTree iniBT(int n) {
    BTree T;
    int m = 0;
    // 生成一个 0 - 9 的随机数
    m = rand() % 10;
    // 如果节点数小于等于 0 或者随机数为 0，返回空树
    if (n <= 0 || m == 0) {
        return NULL;
    }
    // 为新节点分配内存
    T = (BTree)malloc(sizeof(btree));
    T->data = m;
    // 递归创建左子树和右子树
    T->lchild = iniBT(n - 1);
    T->rchild = iniBT(n - 1);
    return T;
}

// 打印二叉树的函数
void print_tree(BTree T, int indent, bool isRightChild = false) {
    if (T == NULL) {
        return;
    }
    // 先递归打印右子树
    print_tree(T->rchild, indent + 4, true);
    // 打印当前节点，通过空格控制缩进
    for (int i = 0; i < indent; i++) {
        std::cout << " ";
    }
    // 根据是否为右子树添加相应标识
    if (isRightChild) {
        std::cout << "├──R─ " << T->data << std::endl;
    } else {
        std::cout << "└──L─ " << T->data << std::endl;
    }
    // 再递归打印左子树
    print_tree(T->lchild, indent + 4, false);
}

int main() {
    BTree T;
    // 初始化一个深度为 3 的二叉树
    T = iniBT(3);
    // 以缩进 10 开始打印二叉树
    print_tree(T, 10);
    return 0;
}

这段 C++代码主要涉及二叉树的创建和打印功能。首先定义了二叉树节点结构体，包含数据域以及左右子树指针。iniBT函数用于创建二叉树，通过传入节点数量参数n，利用随机数生成来决定是否创建节点，若创建则为节点赋值并递归创建左右子树。print_tree函数用于打印二叉树，采用递归的方式，先打印右子树，再打印当前节点，通过控制缩进和添加左右子树标识来呈现类似树枝的效果，最后打印左子树。在main函数中，先创建一个深度为 3 的二叉树，然后调用print_tree函数进行打印。