二叉树的结构体一般定义为:一个指向左子树的指针,一个指向右子树的指针,再加上一个int 或者char 来存储数据,数据结构如下:
typedef struct Node{
struct Node lk; //左子树
struct Node rk; //右子树
int num;
}node, *tree;
首先先来一个函数,往二叉树里添加数据:
tree add_node(tree lk, tree rk, int num)
{
tree new_node = (tree)malloc(sizeof(node));
new_node->lk = lk;
new_node->rk = rk;
new_node->num = num;
}
其中,左右子树都可以是null,不影响。
现在假设我们已经有一个现成的二叉树结构了,其根节点为root,现在来写一个计算叶子节点数和总节点数的函数,首先先来了解一下叶子和根长啥样:
二叉树的叶子节点是没有子树的,而根节点可以有一到两个子树。
计算总节点的个数的代码:
int num_nodes(tree root)
{
if (root == NULL)
return 0;
return 1 + num_nodes(root->lk) + num_nodes(root->rk);
}
计算叶子节点的个数的代码:
int leaf(tree root)
{
if (root == NULL)
return 0;
if (root->lk == NULL && root->rk == NULL)
return 1;
return leaf(root->lk) + leaf(root->rk);
}
对比上面两个代码,可以发现叶子节点的代码就是在总节点代码的基础上加了一个条件判断语句,只要是叶子,那么总数就加一,而总节点的逻辑是:把整颗二叉树遍历一遍,只要不是空节点,那么总数加一。
上求二叉树高的代码:
int height(tree root)
{
if (root==NULL)
return 0;
int hl = height(root->lk);
int hr = height(root->rk);
return 1+(hl>hr?hl:hr);
}
最后,按前序遍历打印出二叉树:
void print_tree(tree root)
{
if (root != NULL){
print("%d ", root->num);
print_tree(root->lk);
print_tree(root->rk);
}
}
根据上图的二叉树,打印出来的遍历顺序为:
1 2 3 4 5,此遍历方法为先遍历完整个左子树(每次都往左边走),再是右子树。