通过用队列完成二叉树的层次遍历,来计算最长宽度。
实现:
用一个字典/hash表来存二叉树中每个节点所在的层数,然后根据所在层数统计该层的节点个数,再比对每层的宽度大小来确定最长宽度。
代码:
static int MaxWidth(Node head)
{
if(head == null)
{
return 0;
}
//队列存二叉树的顺序遍历
Queue<Node> queue = new Queue<Node>();
queue.Enqueue(head);
//字典存节点所对应层数
Dictionary<Node, int> levelMap = new Dictionary<Node, int>();
levelMap.Add(head, 1);
int curLevel = 1;
int curLevelNodes = 0;
int max = int.MinValue;
//当队列不空
while (queue.Count > 0)
{
Node cur = queue.Dequeue();
int curNodeLevel = levelMap[cur];
if(curNodeLevel == curLevel)
{
//如果是当前层就计数
curLevelNodes++;
}
else
{
max = Math.Max(curLevelNodes, max);
curLevel++;
//如果不是当前层,则表示是下一层的第一个,初始值为1
curLevelNodes = 1;
}
if (cur.left != null)
{
//然后子节点的层数记录在字典里
levelMap.Add(cur.left, curLevel + 1);
queue.Enqueue(cur.left);
}
if (cur.right != null)
{
levelMap.Add(cur.right, curLevel + 1);
queue.Enqueue(cur.right);
}
}
max = Math.Max(curLevelNodes, max);
return max;
}