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

解答方法分析:

  1. 初始化一个队列来进行层次遍历,并将根节点入队。
  2. 定义maxSum和maxLevel,并给它们设定一个初始值。
  3. 进入循环,直到队列为空,每次循环遍历当前层的节点。
  4. 在循环中,需要记录当前层节点的总和,并在遍历过程中更新总和。
  5. 如果当前层节点的总和大于maxSum,则更新maxSum和maxLevel。
  6. 循环结束后,返回maxLevel作为结果,即具有最大总和的层数。

所用编程语言: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 int整型
     */
    int maxLevelSum(TreeNode* root) {
        if (root == nullptr) {
            return 0;
        }

        queue<TreeNode*> q;
        q.push(root);
        int maxSum = INT_MIN;
        int maxLevel = 0;
        int level = 1;

        while (!q.empty()) {
            int size = q.size();
            int sum = 0;

            for (int i = 0; i < size; i++) {
                TreeNode* node = q.front();
                q.pop();

                sum += node->val;

                if (node->left != nullptr) {
                    q.push(node->left);
                }

                if (node->right != nullptr) {
                    q.push(node->right);
                }
            }

            if (sum > maxSum) {
                maxSum = sum;
                maxLevel = level;
            }

            level++;
        }

        return maxLevel;
    }

};