题目:从上到下按层打印二叉树,同一层的结点按从左到右的顺序打印,每一层打印到一行。例如,打印下图中的二叉树的结果为:
8
6 10
5 7 9 11
思路:前面讲过如何从上往下按层打印二叉树,这里需要加入换行问题。引入两个变量:一个变量表示在当前层中还没有打印的结点数,另一个变量表示下一层结点的数目。
void Print(BinaryTreeNode* pRoot)
{
if(pRoot == NULL)
return;
std:queue<BinaryTreeNode*> nodes;
nodes.push(pRoot);
int nextLevel = 0;
int toBePrinted = 1;
while(!nodes.empty())
{
BinaryTreeNode* pNode = nodes.front();
printf("%d ", pNode->m_nValue);
if(pNode->m_pLeft != NULL)
{
nodes.push(pNode->m_pLeft);
++nextLevel;
}
if(pNode->m_pRight != NULL)
{
nodes.push(pNode->m_pRight);
++nextLevel;
}
nodes.pop();
--toBePrinted;
if(toBePrinted == 0)
{
printf("\n");
toBePrinted = nextLevel;
nextLevel = 0;
}
}
}