LeetCode 662. 二叉树最大宽度

最新推荐文章于 2022-08-27 13:53:24 发布

interstellar-ai

最新推荐文章于 2022-08-27 13:53:24 发布

阅读量112

点赞数

分类专栏： LeetCode 文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/m0_38128647/article/details/126151580

版权

LeetCode 专栏收录该内容

79 篇文章 0 订阅

订阅专栏

二叉树最大宽度

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

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

在这里插入图片描述

class Solution {
public:
    int widthOfBinaryTree(TreeNode* root) {
        queue<pair<TreeNode*, unsigned long long>> que;
        que.push(make_pair(root, 0));
        unsigned long long maxWidth = 0;
        while (que.size()) {
            int n = que.size();
            vector<unsigned long long> allPos;
            for (int i = 0; i < n; ++i) {
                auto front = que.front();
                que.pop();
                allPos.emplace_back(front.second);
                if (front.first->left) {
                    que.push(make_pair(front.first->left, front.second * 2 + 1));
                }
                if (front.first->right) {
                    que.push(make_pair(front.first->right, front.second * 2 + 2));
                }
            }
            maxWidth = max(maxWidth, allPos.back() - allPos.front() + 1);
        }
        return maxWidth;
    }
};