662. 二叉树最大宽度

二叉树最大宽度算法

题目:662. 二叉树最大宽度

思路:宽度是同一层边缘节点之间的长度,所以需要记录节点的位置(宽度是在 int 范围内的,树可能是无限长的,,,试探 unsigned long long 可以或者计算子节点 pos 减去首节点的值从 0开始

dfs 记录最左非空节点位置

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    void dfs(TreeNode* root, int height, unsigned long long pos, int &maxn, unordered_map<int, int> &height_left) {
        if (root == nullptr) {
            return;
        }
        if (!height_left.count(height)) {
            height_left[height] = pos;
        }
        maxn = max(maxn, static_cast<int>(pos - height_left[height] + 1));
        dfs(root->left, height + 1, pos*2, maxn, height_left);
        dfs(root->right, height + 1, pos*2 + 1, maxn, height_left);
    }
     int widthOfBinaryTree(TreeNode* root) {
        unordered_map<int, int> height_left;
        int maxn = 0;
        dfs(root, 1, 1, maxn, height_left);
        return maxn;
    }
};

bfs 非空节点入队,记录节点位置

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
     int widthOfBinaryTree(TreeNode* root) {
        int maxn = 0;
        queue<pair<TreeNode*, unsigned long long>> q;
        if (root) {
            q.push({root, 1});
        }
        while (!q.empty()) {
            auto st = q.front();
            // q.pop();
            int level_size = q.size();
            for (int i = 0; i < level_size; ++i) {
                auto tmp = q.front();
                q.pop();
                maxn = max(maxn, static_cast<int>(tmp.second - st.second + 1));
                if (tmp.first->left) {
                    q.push({tmp.first->left, tmp.second * 2});
                }
                if (tmp.first->right) {
                    q.push({tmp.first->right, tmp.second * 2 + 1});
                }
            }
        }
        return maxn;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值