【重点】【二叉树】662.二叉树最大宽度

Allenlzcoder

已于 2024-01-19 17:06:03 修改

阅读量524

点赞数 6

分类专栏： labuladong的算法小抄笔记文章标签： BFS 二叉树的最大宽度

于 2024-01-19 14:27:59 首次发布

本文链接：https://blog.csdn.net/Allenlzcoder/article/details/135697242

版权

labuladong的算法小抄笔记专栏收录该内容

77 篇文章 0 订阅

订阅专栏

本文介绍了两种方法解决二叉树宽度问题：一种是修改原TreeNode结构，另一种是构建新TreeNode。两种方法都使用BFS遍历，通过广度优先搜索计算二叉树的最大宽度。

摘要由CSDN通过智能技术生成

题目

法1：BFS，最佳方法

在这里插入图片描述

版本1：修改原TreeNode

class Solution {
    public int widthOfBinaryTree(TreeNode root) {
        int ans = 0;
        Deque<TreeNode> deque = new LinkedList<>();
        root.val = 1;
        deque.offer(root);
        while(!deque.isEmpty()) {
            int count = deque.size(), startIndex = -1, endIndex = -1;
            for (int i = 0; i < count; i++) {
                TreeNode node = deque.poll();
                endIndex = node.val;
                if (startIndex == -1) {
                    startIndex = node.val;
                }
                if (node.left != null) {
                    node.left.val = node.val * 2;
                    deque.offer(node.left);
                }
                if (node.right != null) {
                    node.right.val = node.val * 2 + 1;
                    deque.offer(node.right);
                }
            }
            ans = Math.max(ans, endIndex - startIndex + 1);
        }

        return ans;
    }
}

版本2：构建新TreeNode

class Solution {
    public int widthOfBinaryTree(TreeNode root) {
        int ans = 0;
        Deque<TreeNode> deque = new LinkedList<>();
        deque.offer(new TreeNode(1, root.left, root.right));
        while(!deque.isEmpty()) {
            int count = deque.size(), startIndex = -1, endIndex = -1;
            for (int i = 0; i < count; i++) {
                TreeNode node = deque.poll();
                endIndex = node.val;
                if (startIndex == -1) {
                    startIndex = node.val;
                }
                if (node.left != null) {
                    deque.offer(new TreeNode(node.val * 2, node.left.left, node.left.right));
                }
                if (node.right != null) {
                    deque.offer(new TreeNode(node.val * 2 + 1, node.right.left, node.right.right));
                }
            }
            ans = Math.max(ans, endIndex - startIndex + 1);
        }

        return ans;
    }
}