思路:
一开始考虑用BFS,但是题目要求null也算在宽度之内,但如果node.left为空是无法入队的,所以就改变思路,将每个结点都对应一个坐标,利用规律根结点下标i,左孩子为2*i,右孩子为2*i+1。
public int widthOfBinaryTree(TreeNode root) {
int res=0;
Queue<TreeNode> queue=new LinkedList<>();
queue.add(root);
while(!queue.isEmpty()){
int count=queue.size();
res=Math.max(res,count);
for(int i=0;i<count;i++){
TreeNode node=queue.poll();
if(node.left!=null) queue.add(node.left);
if(node.right!=null) queue.add(node.right);
}
}
return res;
}
为了得到每一行的第一个和最后一个结点,使用双向队列。
public int widthOfBinaryTree(TreeNode root) {
if(root==null) return 0;
Deque<TreeNode> queue=new LinkedList<>();
queue.add(root);
root.val=0;
int max=0;
while(!queue.isEmpty()){
max=Math.max(max,queue.getLast().val-queue.getFirst().val+1);
for(int i=queue.size();i>0;i--){
TreeNode node=queue.poll();
int val=node.val;
if(node.left!=null){
node.left.val=2*val;
queue.add(node.left);
}
if(node.right!=null){
node.right.val=2*val+1;
queue.add(node.right);
}
}
}
return max;
}