考察的知识点:二叉树的层次遍历;
解答方法分析:
- 初始化一个队列来进行层次遍历,并将根节点入队。
- 定义maxSum和maxLevel,并给它们设定一个初始值。
- 进入循环,直到队列为空,每次循环遍历当前层的节点。
- 在循环中,需要记录当前层节点的总和,并在遍历过程中更新总和。
- 如果当前层节点的总和大于maxSum,则更新maxSum和maxLevel。
- 循环结束后,返回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; } };