题解 | #重量级的一层#
重量级的一层
https://www.nowcoder.com/practice/193372871b09426ab9ea805f0fd44d5c
考察的知识点:二叉树的层次遍历;
解答方法分析:
- 初始化一个队列来进行层次遍历,并将根节点入队。
- 定义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;
}
};
查看13道真题和解析