一、题目描述
给定一个二叉树,编写一个函数来获取这个树的最大宽度。树的宽度是所有层中的最大宽度。这个二叉树与**满二叉树(full binary tree)**结构相同,但一些节点为空。
每一层的宽度被定义为两个端点(该层最左和最右的非空节点,两端点间的null节点也计入长度)之间的长度。
示例 1:
示例 2:
示例 3:
示例 4:
二、解题
BFS
这个二叉树与满二叉树结构相同,所以我们将这个二叉树重新赋值,然后计算二叉树的节点值,使用双端队列,队尾元素减队头元素获取最大长度!
class Solution {
public int widthOfBinaryTree(TreeNode root) {
//双端队列
if(root == null){
return 0;
}
Deque<TreeNode> queue = new LinkedList<>();
root.val = 1;
queue.add(root);
int res = Integer.MIN_VALUE;
while(!queue.isEmpty()){
int size = queue.size();
//队列结尾减去开头的值加一即为当前层的宽度
res = Math.max(res,queue.getLast().val - queue.getFirst().val + 1);
for(int i = 0;i<size;i++){
TreeNode node = queue.poll();
if(node.left != null){
node.left.val = node.val * 2;
queue.add(node.left);
}
if(node.right != null){
node.right.val = node.val*2 + 1;
queue.add(node.right);
}
}
}
return res;
}
}
- 时间复杂度:O(N)
- 空间复杂度:O(N)