二叉树的最大宽度

public class TreeMaxWidth {
    public static class Node{
        public int val;
        public Node left;
        public Node right;

        public Node(int data){
            this.val = data;
        }
    }
    public static int maxWidthUseMap(Node head){
        if(head == null){
            return 0;
        }
        Queue<Node> queue = new LinkedList<>();
        queue.add(head);
        HashMap<Node,Integer> levelMap = new HashMap<>();//哈希表记录该节点是在哪一层
        levelMap.put(head,1);
        int curLevel = 1;//当前你正在统计哪一层的宽度
        int curLevelNodes = 0;//当前层curLevel的宽度是多少
        int max = 0;
        while(!queue.isEmpty()){
            Node cur = queue.poll();
            int curNodeLevel = levelMap.get(cur);//该节点是在哪一层
            if(curNodeLevel == curLevel){//如果该节点是再统计的那一层，就把宽度加加
                curLevelNodes++;
            }else{//如果该节点不在统计的那一层，说明已经来到了下一层，那么上一层就该结算了,开始下层
                max = Math.max(max,curLevelNodes);
                curLevel++;
                curLevelNodes = 1;
            }
            //要保证哈希表中记录节点是在哪一层
            if(cur.left != null){
                levelMap.put(cur.left,curNodeLevel + 1);
                queue.add(cur.left);
            }
            if(cur.right != null){
                levelMap.put(cur.right,curNodeLevel + 1);
                queue.add(cur.right);
            }
        }
        return max;
    }
}