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