给定一个非空二叉树, 返回一个由每层节点平均值组成的数组

给定一个非空二叉树, 返回一个由每层节点平均值组成的数组。
(节点值的范围在32位有符号整数范围内)

输入:

[3,9,20,15,7]

输出:

[3.00000,14.50000,11.00000]

class Solution {
    double[] a=new double[1000];
    int[] b=new int[1000];
    int count=0;
    public List<Double> averageOfLevels(TreeNode root) {
        List<Double> list=new ArrayList<Double>();
        dfs(root,0);
        for(int i=0;i<=count;i++){
            list.add(a[i]*1.0/b[i]);
        }
        return list;
    }
    public void dfs(TreeNode r,int k){
        if(r.left==null&&r.right==null){
            a[k]+=r.val;
            b[k]++;
            count=k>count?k:count;
        }else{
            a[k]+=r.val;
            b[k]++;
            if(r.right!=null){
                dfs(r.right,k+1);
            }
            if(r.left!=null){
                dfs(r.left,k+1);
            }
        }
    }
}

//leetcode提交执行用时内存消耗
如果你的提交不尽人意,建议多提交几次—#滑稽
在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 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、付费专栏及课程。

余额充值