662. 二叉树最大宽度
给定一个二叉树,编写一个函数来获取这个树的最大宽度。树的宽度是所有层中的最大宽度。这个二叉树与满二叉树(full binary tree)结构相同,但一些节点为空。
每一层的宽度被定义为两个端点(该层最左和最右的非空节点,两端点间的
null
节点也计入长度)之间的长度。示例 1:
输入: 1 / \ 3 2 / \ \ 5 3 9 输出: 4 解释: 最大值出现在树的第 3 层,宽度为 4 (5,3,null,9)。
示例 2:
输入: 1 / 3 / \ 5 3 输出: 2 解释: 最大值出现在树的第 3 层,宽度为 2 (5,3)。
示例 3:
输入: 1 / \ 3 2 / 5 输出: 2 解释: 最大值出现在树的第 2 层,宽度为 2 (3,2)。
示例 4:
输入: 1 / \ 3 2 / \ 5 9 / \ 6 7 输出: 8 解释: 最大值出现在树的第 4 层,宽度为 8 (6,null,null,null,null,null,null,7)。
注意: 答案在32位有符号整数的表示范围内。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int widthOfBinaryTree(TreeNode* root) {
if (!root) return 0;
int res = 0;
queue<pair<TreeNode*, unsigned long>> q1, q2;
q1.push({root, 1});
while (!q1.empty()) {
q1.swap(q2);
int left = q2.front().second;
pair<TreeNode*, unsigned long> temp;
while (!q2.empty()) {
temp = q2.front();
q2.pop();
if (temp.first->left) q1.push({temp.first->left, temp.second * 2});
if (temp.first->right) q1.push({temp.first->right, temp.second * 2 + 1});
}
int right = temp.second;
res = max(res, right - left + 1);
}
return res;
}
};
层序遍历,同时记录下每个结点的编号(根为1,就像在堆中的那样),每层首尾元素的编号差+1就是当前层的宽度。
2020/07/05 12:24