【leetcode】二叉树最大宽度

最新推荐文章于 2022-08-27 10:59:05 发布

William_Chan_6

最新推荐文章于 2022-08-27 10:59:05 发布

阅读量180

点赞数

分类专栏： leetcode 文章标签： leetcode c++ 算法数据结构

本文链接：https://blog.csdn.net/weixin_44340194/article/details/108358044

版权

leetcode 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

原题链接：leecode原题

算法关键：
在这里插入图片描述
（以上摘自leetcode题解）

样例代码：摘自leetcode题解

class Solution {
public:
    int widthOfBinaryTree(TreeNode* root) {
        if (!root) return 0;
        queue<pair<TreeNode*, unsigned long long>> q;
        int ans = 1;
        q.push({root, 1});
        while (!q.empty()) {
            int sz = q.size();
            ans = max(int(q.back().second - q.front().second + 1), ans);
            for (int i=0; i < sz; i++) {
                TreeNode *node = q.front().first;
                unsigned long long pos = q.front().second;
                q.pop();
                if (node->left) q.push({node->left, 2 * pos});
                if (node->right) q.push({node->right, 2 * pos + 1});
            }
        }
        return ans;
    }
};