题解 | #重量级的一层#
重量级的一层
https://www.nowcoder.com/practice/193372871b09426ab9ea805f0fd44d5c
知识点:
二叉树/队列
分析:
层序遍历二叉树,然后更新每一层的和,找出最大的那一层即可。
完整代码C++——
class Solution { public: /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param root TreeNode类 * @return int整型 */ int maxLevelSum(TreeNode* root) { if(root == nullptr) return 0; queue<TreeNode*> q; q.push(root); int level = 0; int cnt = 0,cnt2 = 0; int ans = 0; while(!q.empty()){ int size = q.size(); while(size --){ auto t = q.front();q.pop(); cnt += t->val; if(t->left) q.push(t->left); if(t->right) q.push(t->right); } level++; if(cnt > cnt2) { cnt2 = cnt; cnt = 0; ans = level; } } return ans; } };