统计二叉树中度为0的结点个数,度为1的结点个数,度为2的结点个数
要统计结点个数,要判断二叉树的左右孩子,对于度为0的结点,左右孩子均不存在,对于度为1的结点,只存在右孩子,或者只存在左孩子,对于度为2的结点,左右孩子均存在,以下是代码实现
#include <iostream>
using namespace std;
struct TreeNode {
int val;//表示节点的值。
TreeNode* left;//指向左子节点的指针。
TreeNode* right;//指向右子节点的指针。
TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};
void countNodes(TreeNode* root, int& count0, int& count1, int& count2) {
if (root == nullptr) {
return;
}
//先判断左右支是否为空
int degree = 0;
if (root->left != nullptr) {
degree++;
}
if (root->right != nullptr) {
degree++;
}
//度为0的结点的个数
if (degree == 0) {
count0++;
}
//度为1的结点的个数
else if (degree == 1) {
count1++;
}
//度为2的结点的个数
else {
count2++;
}
//再次判断左子树和右子树的度的情况
countNodes(root->left, count0, count1, count2);
countNodes(root->right, count0, count1, count2);
}
int main() {
TreeNode* root = new TreeNode(1);
root->left = new TreeNode(2);
root->right = new TreeNode(3);
root->left->left = new TreeNode(4);
root->left->right = new TreeNode(5);
root->right->left = new TreeNode(6);
int count0 = 0, count1 = 0, count2 = 0;
countNodes(root, count0, count1, count2);
cout << "度为0的结点个数:" << count0 << endl;
cout << "度为1的结点个数:" << count1 << endl;
cout << "度为2的结点个数:" << count2 << endl;
return 0;
}
struct TreeNode 是一个自定义的数据结构,用于表示二叉树中的节点。它包含以下成员变量:
int val:表示节点的值。
TreeNode* left:指向左子节点的指针。
TreeNode* right:指向右子节点的指针。
在这个结构体中,使用构造函数TreeNode(int x) 来初始化节点的值,并将左右子节点的指针初始化为空指针(nullptr)。
通过定义这样一个结构体,我们可以创建二叉树,并通过连接节点的指针来表示树的结构。在处理二叉树相关问题时,我们可以使用TreeNode结构体来创建和操作树的节点。
TreeNode* root = new TreeNode(1);
这行代码创建了一个名为root的指针,它指向一个新创建的TreeNode对象。此节点的值(val)被设置为1,并且左右子节点的指针被初始化为nullptr。
root->left = new TreeNode(2);
这行代码创建了一个新的TreeNode对象,其值为2,并将其赋给root节点的左子节点(left)。
root->right = new TreeNode(3);
这行代码创建了一个新的TreeNode对象,其值为3,并将其赋给root节点的右子节点(right)。
root->left->left = new TreeNode(4);
这行代码创建了一个新的TreeNode对象,其值为4,并将其赋给root节点的左子节点的左子节点。
root->left->right = new TreeNode(5);
这行代码创建了一个新的TreeNode对象,其值为5,并将其赋给root节点的左子节点的右子节点。
root->right->left = new TreeNode(6);
这行代码创建了一个新的TreeNode对象,其值为6,并将其赋给root节点的右子节点的左子节点。
通过这些代码,我们创建了如下的二叉树结构:
1
/ \
2 3
/ /
4 6
\
5
每个节点的值和子节点关系可以通过指针来表示。这个二叉树有特定的层次结构,可以在之后的代码中使用和操作这个二叉树。