1161. 最大层内元素和
题目链接:1161. 最大层内元素和
代码如下:
/*** Definition for a binary tree node.* struct TreeNode {* int val;* TreeNode *left;* TreeNode *right;* TreeNode() : val(0), left(nullptr), right(nullptr) {}* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}* };*/
class Solution
{
public://层次遍历int maxLevelSum(TreeNode* root) {queue<TreeNode*> que;//level代表目前所在树的层数,levelSum代表目前树层的总和int res=1,level=1,levelSum=INT_MIN;que.push(root);while(!que.empty()){int len=que.size();int tmpSum=0;for(int i=0;i<len;i++)//遍历此层{TreeNode* tmp=que.front();que.pop();tmpSum+=tmp->val;if(tmp->left) {que.push(tmp->left);}if(tmp->right) {que.push(tmp->right);}}if(levelSum<tmpSum)//更新结果{levelSum=tmpSum;res=level;}level++;//记录下一层}return res;}
};