给定一个二叉树,编写一个函数来获取这个树的最大宽度。树的宽度是所有层中的最大宽度。这个二叉树与满二叉树(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位有符号整数的表示范围内。
思路:记一下每层开始和结尾就好啦
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
class Solution {
int[] a;
int[] b;
public int widthOfBinaryTree(TreeNode root) {
if(root==null) return 0;
if(root.left==null && root.right==null)
return 1;
a=new int[10000];
b=new int[10000];
for(int i=0;i<10000;i++) a[i]=-1;
dfs(root,0,0);
int max=0;
for(int i=0;i<10000;i++)
max=Math.max(max, b[i]-a[i]);
return max;
}
private void dfs(TreeNode root,int depth,int sum) {
if(root==null) return;
if(a[depth]==-1) a[depth]=sum;
b[depth]=sum+1;
dfs(root.left,depth+1,sum*2);
dfs(root.right,depth+1,sum*2+1);
}
}