662. 二叉树最大宽度

最新推荐文章于 2024-07-22 16:27:42 发布

overdosed_

最新推荐文章于 2024-07-22 16:27:42 发布

阅读量290

点赞数

分类专栏： leetcode 文章标签： leetcode 算法 java

本文链接：https://blog.csdn.net/weixin_45904682/article/details/130009975

版权

leetcode 专栏收录该内容

82 篇文章 0 订阅

订阅专栏

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