二叉树的宽度

2024-05-24 14:29:48 浏览数 (2)

题目: 给定一颗二叉树,求二叉树的宽度。

宽度的定义: 二叉树的宽度定义为具有最多结点数的层中包含的结点数。

比如上图中,第 1 层有 1 个结点, 第 2 层有 2 个结点, 第 3 层有 4 个结点, 第 4 层有 1 个结点。可知,第 3 层的结点数最多,所以这棵二叉树的宽度是 4。

求解思路: 这里需要用到二叉树的层次遍历,即广度优先周游。在层次遍历的过程中,通过读取队列中保留的上一层的结点数来记录每层的结点数,以获取所有层中最大的结点数。关于二叉树广度优先周游,参考:二叉树的遍历。

具体实现:

代码语言:javascript复制
//求二叉树的宽度  
int treeWidth(BinaryTreeNode *pRoot) {
    if (pRoot == NULL) return 0;
    
    queue<BinaryTreeNode*> myQueue;  
    myQueue.push(pRoot);		// 将根结点入队列
	int nWidth = 1;				// 二叉树的宽度
    int nCurLevelWidth = 1;		// 记录当前层的宽度
    BinaryTreeNode *pCur = NULL;
  
	// 队列不为空
    while (!myQueue.empty()) {
        while (nCurLevelWidth != 0) {
            pCur = myQueue.front();	// 取出队首元素
            myQueue.pop();			// 队首元素出队列
  
            if (pCur->m_pLeft != NULL) myQueue.push(pCur->m_pLeft);
            if (pCur->m_pRight != NULL) myQueue.push(pCur->m_pRight);
            nCurLevelWidth--;
        }
  
        nCurLevelWidth = myQueue.size();
        nWidth = nCurLevelWidth > nWidth ? nCurLevelWidth : nWidth;
    }
    return nWidth;  
}

参考文献

[1] 何海涛.剑指Offer[M].电子工业出版社. [2] CSDN.求二叉树的深度和宽度

0 人点赞