题目描述:
给定一个二叉树,编写一个函数来获取这个树的最大宽度。树的宽度是所有层中的最大宽度。这个二叉树与满二叉树(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)。
解题思路:将二叉树看成一个完全二叉树,在完全二叉树中,如果一个父亲结点是i,则他的左右孩子结点分别是:2*i,2*i+1,然后利用深搜进行比较,就可以找出最大值,第一遍深搜是一个找到深度的过程,之后就是比较。
/** 假设满二叉树表示成数组序列, 根节点所在的位置为1, 则任意位于i节点的左右子节点的index为2*i, 2*i+1 用一个List保存每层的左端点, 易知二叉树有多少层,List的元素就有多少个. 那么可以在dfs的过程中记录每个 节点的index及其所在的层level, 如果level > List.size()说明当前节点就是新的一层的最左节点, 将其 加入List中, 否则判断当前节点的index减去List中对应层的最左节点的index的宽度是否大于最大宽度并更新 **/
public class Solution1 {
private int max = 0;
public int widthOfBinaryTree(TreeNode root) {
dfs(root,1,1,new ArrayList<>());
return max;
}
public void dfs(TreeNode node, int index, int level, List<Integer> left){
if(node == null)
return;
if(level>left.size()) left.add(index);
max = Math.max(max,index-left.get(level-1)+1);
dfs(node.left,index*2,level+1,left);
dfs(node.right,index*2+1,level+1,left);
}
}