#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
函数进行打印。