题目描述
给定一个二叉树,编写一个函数来获取这个树的最大宽度。树的宽度是所有层中的最大宽度。这个二叉树与满二叉树(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位有符号整数的表示范围内。
定义一开始的头结点编号为1,其左右儿子节点编号分别为 i * 2,i * 2+1,以此类推。要求解的是树的宽度,也就是每一层中最左边和最右边节点的差值,由于有可能出现树的深度很高,每次去做乘法的时候会超出数据范围,将每层的节点全部转换为从1开始的序号,转化的方式就是先记录本层的节点最左边的编号 l ,本层节点的编号转换就是本层节点的编号减去之前记录的 l 加1 ,然后不断更新本层最右边的节点编号 r ,本层的宽度计算为 r-l+1,下一层的节点编号分别为本层节点编号的2倍,以及2倍加1,下一层节点的转化也类似。
class Solution {
public:
int widthOfBinaryTree(TreeNode* root) {
queue<pair<TreeNode*,int>> q;
q.push({root,1});
int res=1;
while(q.size())
{
int n=q.size();
int l=q.front().second,r;
for(int i=0;i<n;i++)
{
auto t=q.front(); q.pop();
auto p=t.second-l+1;//本层节点的编号
auto v=t.first;
r=t.second;//不断更新本层的右边的节点编号
if(v->left) q.push({v->left,p*2});
if(v->right) q.push({v->right,p*2+1});
}
res=max(res,r-l+1);
}
return res;
}
};