二叉树最大宽度

最新推荐文章于 2024-07-21 22:06:29 发布

luck++

最新推荐文章于 2024-07-21 22:06:29 发布

阅读量391

点赞数 1

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

本文链接：https://blog.csdn.net/qq_61344207/article/details/130540757

版权

算法专栏收录该内容

14 篇文章 0 订阅

订阅专栏

Width first traversal

数据

使用变量curLevel 记录当前遍历层数；curLevelnode 记录当前层的节点个数

哈希表Level记录节点与它所在层数的对应关系

队列NodeSeq实现层序遍历

思路

将根节点和它所在第一层入哈希表Level

根节点入队列NodeSeq，进行层序遍历

每次pop节点Front，比较Front所在层数与curLevel

实现

void WidthMax(Node* root)
{
    if(root != nullptr)
    {
        queue<Node*> NodeSeq;
        int curLevel = 1;
        int curLevelnode = 0;
        unordered_map<Node*,int> Level;
        int width = -1;
        
        Level[root] = 1;
        NodeSeq.push(root);
        
        while(!NodeSeq.empty())
        {
            Node* Front = NodeSeq.front();
            NodeSeq.pop();
            if(Level[Front] == curLevel)
            {
                curLevelnode++;
            }
            else
            {
                width = max(width,curLevelnode);
                curLevel++;
                curLevelnode = 1;
            }
            
            if(Front->left) 
            {
                NodeSeq.push(Front->left);
                Level[Front->left] = curLevel + 1;
            }
            if(Front->right)
            {
                NodeSeq.push(Front->right);
                Level[Front->right] = curLevel + 1;
            }
        }
        cout << width << endl;
    }
}