LC662.二叉树最大宽度

662. 二叉树最大宽度

给定一个二叉树，编写一个函数来获取这个树的最大宽度。树的宽度是所有层中的最大宽度。这个二叉树与满二叉树（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位有符号整数的表示范围内。

题解：
使用宽度优先遍历，记录下每个非null节点的位置，然后使用最右的节点的位置减去最左节点的位置就是这一层的宽度

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    /**
    使用宽度优先遍历，记录下每个非null节点的位置，然后使用最右的节点的位置减去最左节点的位置就是这一层的宽度
     */
    public int widthOfBinaryTree(TreeNode root) {
        Deque<AnnotateNode> deque=new LinkedList<>();
        deque.addLast(new AnnotateNode(root,0));
        int width=0;
        while(!deque.isEmpty()){
            int len=deque.size();
            int left=deque.peekFirst().pos;
            int right=deque.peekLast().pos;
            if(right-left+1>width){
                width=right-left+1;
            }
            for(int i=0;i<len;i++){
                AnnotateNode n=deque.pollFirst();
                if(n.node!=null){
                    if(n.node.left!=null){
                        deque.addLast(new AnnotateNode(n.node.left,n.pos*2));
                    }
                    if(n.node.right!=null){
                        deque.addLast(new AnnotateNode(n.node.right,n.pos*2+1));
                    }
                }
            }
        }
        return width;
    }

    class AnnotateNode{
        TreeNode node;
        int pos;
        public AnnotateNode(TreeNode node,int pos){
            this.node=node;
            this.pos=pos;
        }
    }
}