662. 二叉树最大宽度-BFS

Posted 2022-06-10 hequnwang10

一、题目描述

给定一个二叉树，编写一个函数来获取这个树的最大宽度。树的宽度是所有层中的最大宽度。这个二叉树与**满二叉树（full binary tree）**结构相同，但一些节点为空。

每一层的宽度被定义为两个端点（该层最左和最右的非空节点，两端点间的null节点也计入长度）之间的长度。

示例 1：

示例 2：

示例 3：

示例 4：

二、解题

BFS

这个二叉树与满二叉树结构相同，所以我们将这个二叉树重新赋值，然后计算二叉树的节点值，使用双端队列，队尾元素减队头元素获取最大长度！

class Solution 
    public int widthOfBinaryTree(TreeNode root) 
        //双端队列
        if(root == null)
            return 0;
        
        Deque<TreeNode> queue = new LinkedList<>();
        root.val = 1;
        queue.add(root);
        int res = Integer.MIN_VALUE;
        while(!queue.isEmpty())
            int size = queue.size();
            //队列结尾减去开头的值加一即为当前层的宽度
            res = Math.max(res,queue.getLast().val - queue.getFirst().val + 1);
            for(int i = 0;i<size;i++)
                TreeNode node = queue.poll();
                if(node.left != null)
                    node.left.val = node.val * 2;
                    queue.add(node.left);
                
                if(node.right != null)
                    node.right.val = node.val*2 + 1;
                    queue.add(node.right);
                
            
        
        return res;
    

• 时间复杂度：O(N)
• 空间复杂度：O(N)