LeetCode-662-二叉树最大宽度

LeetCode-662-二叉树最大宽度

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

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

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

题目数据保证答案将会在 32 位 带符号整数范围内。

示例 1：

输入：root = [1,3,2,5,3,null,9]
输出：4
解释：最大宽度出现在树的第 3 层，宽度为 4 (5,3,null,9) 。

示例 2：

输入：root = [1,3,2,5,null,null,9,6,null,7]
输出：7
解释：最大宽度出现在树的第 4 层，宽度为 7 (6,null,null,null,null,null,7) 。

示例 3：

输入：root = [1,3,2,5]
输出：2
解释：最大宽度出现在树的第 2 层，宽度为 2 (3,2) 。

提示：

• 树中节点的数目范围是 [1, 3000]
• -100 <= Node.val <= 100

class Solution {
    /**
     * 在数组中父亲节点与孩子节点的关系 左孩子: index = 2 * i, 右孩子: index = 2 * i + 1
     * 进行层次遍历, 在遍历过程中需要同时记录每一个结点的下标
     */
    /**
     *          1
     *         2 3
     *       4 5 6 7
     * 8 9 10 11 12 13 14 15
     */
    public int widthOfBinaryTree(TreeNode root) {
        Deque<Pair<TreeNode, Integer>> queue = new ArrayDeque<>();
        int width = 0;
        queue.add(new Pair<>(root, 0));
        while (!queue.isEmpty()) {
            int size = queue.size();
            width = Math.max(width, queue.getLast().getValue() - queue.getFirst().getValue() + 1);
            for (int i = 0; i < size; i++) {
                Pair<TreeNode, Integer> pair = queue.poll();
                TreeNode node = pair.getKey();
                if (node.left != null) {
                    queue.add(new Pair<>(node.left, 2 * pair.getValue()));
                }
                if (node.right != null) {
                    queue.add(new Pair<>(node.right, 2 * pair.getValue() + 1));
                }
            }
        }
        return width;
    }

    public static void main(String[] args) {
        Solution solution = new Solution();
        TreeNode root = new TreeNode();
        root.left = new TreeNode();
        root.right = new TreeNode();
        root.left.left = new TreeNode();
        root.right.right = new TreeNode();
        solution.widthOfBinaryTree(root);
    }
    
    
    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;
        }
	}
}