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;
}
};