考察的知识点:二叉树的层次遍历;

解答方法分析:

  1. 首先检查根节点是否为空,如果为空,直接返回空的结果向量。
  2. 创建一个队列,并将根节点入队。
  3. 进入一个循环,每次循环一层节点:获取当前层的节点个数,通过记录队列长度 len实现。使用一个变量 sum 来保存当前层节点值的和。依次从队列中弹出当前层的节点,将其值加到 sum 中,并将其左子节点和右子节点(如果存在)入队。
  4. 计算当前层的平均值,即将 sum 除以 len,并将结果加入到结果向量中。
  5. 循环结束后,返回结果向量作为函数的返回值。

所用编程语言:C++;

完整编程代码:↓

/**
 * struct TreeNode {
 *  int val;
 *  struct TreeNode *left;
 *  struct TreeNode *right;
 *  TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
class Solution {
  public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param root TreeNode类
     * @return double浮点型vector
     */
    vector<double> averageOfLevels(TreeNode* root) {
        vector<double> res;
        if (root == nullptr) {
            return res;
        }

        queue<TreeNode*> queue;
        queue.push(root);
        while (!queue.empty()) {
            int len = queue.size();
            double sum = 0;
            for (int i = 0; i < len; i++) {
                TreeNode* temp = queue.front();
                queue.pop();
                sum += temp->val;
                if (temp->left != nullptr) {
                    queue.push(temp->left);
                }
                if (temp->right != nullptr) {
                    queue.push(temp->right);
                }
            }
            res.push_back(sum / len);
        }

        return res;
    }
};