637.二叉树每层的平均值

这也是个层序遍历的问题

只需要对输出结果求和就可以了,注意求和时要用long类型,不然会过不了大数字的测试

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public static List<List<Integer>> list=new ArrayList<>();
    public static List<Double> averageOfLevels(TreeNode root) {
        selfTree(root,0);
        List<Double> result=new ArrayList<>();
        for (List<Integer> integers : list) {
            long sum=0;
            int num=0;
            for (Integer integer : integers) {
                sum+=integer;
                num++;
            }
            result.add(Double.valueOf((double) sum /num));
        }
        list=new ArrayList<>();
        return result;
    }
    public static void selfTree(TreeNode root,int deep){
        if(root==null)return;
        deep++;
        if(list.size()<deep){
            List<Integer> cur=new ArrayList<>();
            list.add(cur);
        }
        list.get(deep-1).add(root.val);
        selfTree(root.left,deep);
        selfTree(root.right,deep);
    }
}

大部分就是普通的层序遍历,没什么特殊的

好的,对于你的问题,我可以给出以下的 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、付费专栏及课程。

余额充值