给定一个完全的二叉树,计算节点的数量。

217 篇文章 0 订阅
174 篇文章 2 订阅

本题源自leetcode 222

-----------------------------------------------------------------------

思路:完全二叉树,的结点个数。1 如果左子树的高度等于右子树的高度。则为完全二叉树。节点个数等于2的树高次幂-1;

否则就递归遍历左子树节点和右子树节点,再加上根节点。

代码:

 int countNodes(TreeNode* root) {
        if(!root)
            return 0;
        TreeNode *l=root;
        TreeNode* r=root;
        int hl=0,hr=0;
        while(l){
            l=l->left;
            hl++;
        }
        while(r){
            r=r->right;
            hr++;
        }
        if(hr == hl){
            return pow(2,hl)-1;
        }
        return countNodes(root->left)+countNodes(root->right)+1;
    }

思路: 分别计算左子树和右子树的左子树的高度。如果相等,则说明左子树是完全二叉树。否则,右子树比左子树低1,右子树为完全二叉树。

代码2:

 int countNodes(TreeNode* root) {
        if(!root)
            return 0;
        int left=getLeftHigh(root->left);
        int right=getLeftHigh(root->right);
        if(left == right)
            return pow(2,left)+countNodes(root->right);  //左子树为完全二叉树
        return pow(2,right)+countNodes(root->left);   //右子树为完全二叉树。且高度比左子树少1
    }
    int getLeftHigh(TreeNode* root){
        if(!root)
            return 0;
        int high=0;
        while(root){
            root=root->left;
            high++;
        }
        return high;
    }


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,对于你的问题,我可以给出以下的 C 语言实现: ```c #include <stdio.h> #include <stdlib.h> // 定义二叉树节点结构体 typedef struct TreeNode { int val; struct TreeNode *left; struct TreeNode *right; } TreeNode; // 定义队列结构体 typedef struct Queue { TreeNode *elements[100]; int front; int rear; } Queue; // 创建一个新的队列 Queue* createQueue() { Queue *queue = (Queue*) malloc(sizeof(Queue)); queue->front = 0; queue->rear = 0; return queue; } // 入队操作 void enqueue(Queue *queue, TreeNode *node) { queue->elements[queue->rear] = node; queue->rear++; } // 出队操作 TreeNode* dequeue(Queue *queue) { TreeNode *node = queue->elements[queue->front]; queue->front++; return node; } // 判断队列是否为空 int isEmpty(Queue *queue) { return queue->front == queue->rear; } // 计算树的深度 int maxDepth(TreeNode* root) { if (root == NULL) { return 0; } else { int leftDepth = maxDepth(root->left); int rightDepth = maxDepth(root->right); return (leftDepth > rightDepth) ? (leftDepth + 1) : (rightDepth + 1); } } // 计算每一层的平均值 double* averageOfLevels(TreeNode* root, int* returnSize) { // 计算树的深度 int depth = maxDepth(root); // 创建结果数组 double *result = (double*) malloc(depth * sizeof(double)); *returnSize = depth; // 创建队列 Queue *queue = createQueue(); // 先将根节点入队 enqueue(queue, root); // 遍历每一层,计算平均值 for (int i = 0; i < depth; i++) { int levelSize = queue->rear - queue->front; double sum = 0; for (int j = 0; j < levelSize; j++) { TreeNode *node = dequeue(queue); sum += node->val; if (node->left != NULL) { enqueue(queue, node->left); } if (node->right != NULL) { enqueue(queue, node->right); } } result[i] = sum / levelSize; } return result; } // 创建新的二叉树节点 TreeNode* createTreeNode(int val) { TreeNode *node = (TreeNode*) malloc(sizeof(TreeNode)); node->val = val; node->left = NULL; node->right = NULL; return node; } // 根据输入创建二叉树 TreeNode* createBinaryTree() { int val; scanf("%d", &val); if (val == -1) { return NULL; } TreeNode *root = createTreeNode(val); root->left = createBinaryTree(); root->right = createBinaryTree(); return root; } int main() { // 根据输入创建二叉树 TreeNode *root = createBinaryTree(); // 计算每一层的平均值 int size; double *result = averageOfLevels(root, &size); // 输出结果 for (int i = 0; i < size; i++) { printf("%.2f ", result[i]); } return 0; } ``` 以上代码中,我们通过层次遍历二叉树的方式,遍历每一层,并计算每一层节点的平均值。在遍历每一层时,我们使用一个变量 `sum` 来记录该层节点的和,然后将这个和除以该层节点数量,就可以得到该层的平均值。同时,我们使用一个队列来实现层次遍历操作。最终,我们将每一层的平均值存储到一个结果数组中,并返回该数组。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值