求二叉树的最大宽度

给你一棵二叉树的根节点 root ，返回树的 最大宽度 。

树的 最大宽度 是所有层中最大的 宽度 。

每一层的 宽度 被定义为该层最左和最右的非空节点（即，两个端点）之间的长度。将这个二叉树视作与满二叉树结构相同，两端点间会出现一些延伸到这一层的 null 节点，这些 null 节点也计入长度。

记录这道题难点：

（1）BFS求解：

难点：

1. 如何对一层中的 null 节点进行遍历处理
2. 计算每层宽度方法

难点分析：使用层序遍历二叉树（BFS）来求解的化，就必须要处理 null 节点带来的问题，因为我们要借助队列 （Queue） 这一数据结构来实现 BFS ，就要处理 队列 不能加添 null 节点的问题。所以如果要使用队列的话，就不能通过每层的队列中的节点数来统计每层的宽度，这样会忽略每层的null 节点 。

解决方法：利用二叉树的父节点和孩子节点的位置关系，为每一个节点编号（对应的 index 值）。父节点编号：index，左孩子编号：( index  * 2 ) + 1，右孩子编号： ( index * 2 ) + 2，这样就不用担心会漏掉 null 节点，最后只需要用每一层的最后一个节点的编号 减去 第一个节点的编号再加1就是每一层的宽度。

那用什么数据结构来存储每个节点对应编号呢，第一反应就是哈希表，但是哈希表是无序的，这样每一层做减法就不能得到正确的宽度，java 中提供了一种类似哈希表的一种数据结构 Pair

Pair和Map共通点

• Pair和Map都是以key,value进行存储

Pair和Map不同点

• Pair通过getKey()/getValue()获取对应的key值和value值，没有添加键值对的操作
• Map是通过get()获取对应的value，通过values()获取所有的value，而且还可以通过put进行新增键值对。
• pair保存的是一对key value，而map可以保存多对key value。

我们可以借助 Pair 来对节点编号进行记录（这里为了加深一下对 Pair 的记忆，也可以尝试使用有序表 TreeMap来实现）

code:

class Solution {
    public int widthOfBinaryTree(TreeNode root) {
        if(root == null){
            return 0;
        }
        List<Integer> list = new ArrayList<>();
        int max = Integer.MIN_VALUE;
        Queue<Pair<TreeNode,Integer>> queue = new LinkedList<>();
        queue.offer(new Pair(root,0));
        while(!queue.isEmpty()){
            int size = queue.size();
            //每一层重新分配一个集合
            list = new ArrayList<>();
            while(size-- > 0){
                Pair<TreeNode,Integer> pair = queue.poll();
                TreeNode node = pair.getKey();
                list.add(pair.getValue());
                if(node.left != null){
                    queue.offer(new Pair(node.left,pair.getValue() * 2 + 1));
                }    
                if(node.right != null){
                    queue.offer(new Pair(node.right,pair.getValue() * 2 + 2));
                }
            }
            //每遍历完一层，就更新最大宽度
            max = Math.max(max,list.get(list.size() - 1) - list.get(0) + 1);
        }
    return max;
    }
}

官方题解也是借助了 Pair 和 List 共同实现了一个有序表，用 List<Pair<TreeNode,Integer>> arr 代替了 Queue ，由于 List 不能 poll() ( 查看并删除队头元素 )，所以通过 List<Pair<TreeNode,Integer>> tmp 来不断更新 arr ,达到 Queue.poll() 的效果。

code:

class Solution {
    public int widthOfBinaryTree(TreeNode root) {
        int res = 1;
        List<Pair<TreeNode, Integer>> arr = new ArrayList<Pair<TreeNode, Integer>>();
        arr.add(new Pair<TreeNode, Integer>(root, 1));
        while (!arr.isEmpty()) {
            List<Pair<TreeNode, Integer>> tmp = new ArrayList<Pair<TreeNode, Integer>>();
            for (Pair<TreeNode, Integer> pair : arr) {
                TreeNode node = pair.getKey();
                int index = pair.getValue();
                if (node.left != null) {
                    tmp.add(new Pair<TreeNode, Integer>(node.left, index * 2));
                }
                if (node.right != null) {
                    tmp.add(new Pair<TreeNode, Integer>(node.right, index * 2 + 1));
                }
            }
            res = Math.max(res, arr.get(arr.size() - 1).getValue() - arr.get(0).getValue() + 1);
            arr = tmp;
        }
        return res;
    }
}