[编程之美][3.10] 分层遍历二叉树

 

　　《编程之美》3.10节中主要针对二叉树的分层遍历进行了讲述，其实在《剑指Offer》一书中”面试题23：从上到下打印二叉树“ 实质上也是针对树结构的按层遍历算法。解法使用一个队列实现。采用递归，代码如下：

struct TreeNode{
    TreeNode *left;
    TreeNode *right;

    int value;
};

void visitTreeByLevel(TreeNode *root){
    if(NULL == root){
        return;
    }

    queue<TreeNode *> visitQue;
    visitQue.push(root);

    while(!visitQue.empty()){
        TreeNode *curNode = visitQue.front();
        if(curNode->left){
            visitQue.push(curNode->left);
        }
        if(curNode->right){
            visitQue.push(curNode->right);
        }
        printf("%d ", curNode->value);
        visitQue.pop();
    }
}

　　而在《编程之美》一书中有更进一步的要求： 树结构每一层需要单独输出一行，因此我们需要标记每一行的起始，以便进行行的结束和下一行数据的开始。因此我们定义两个标示记录每一行的开始和结束，而每次每一行遍历结束的时候实际上已经取得了此行的元素个数。那么这样自然而然我们使用该行元素个数为结束标记，并且需要在遍历完毕每一行数据之后重新设置标示。 代码如下：

void visitTreeByLevel(TreeNode*  pRoot){
    if (NULL == pRoot){
        return;
    }

    queue<TreeNode*> visitQue;
    visitQue.push(pRoot);

    while (visitQue.size()>0){
        int nBegin = 0;
        int nEnd = visitQue.size();
        
        while (nBegin < nEnd){        // 每次将该层全部便利完毕之后 再设置新的标识
            TreeNode* pElem = visitQue.front();
            if (pElem->left){
                visitQue.push(pElem->left);
            }

            if (pElem->right){
                visitQue.push(pElem->right);
            }

            cout<<pElem->value<<" ";
            nBegin++;                // 追赶end
            visitQue.pop();

            if (nBegin == nEnd){
                cout<<"\t\t\t该层有"<<nEnd<<"个元素！";
            }
        }
        cout<<endl;                    // 每一行输出换行
    }
}

 

希望各位看官不吝赐教，小弟感谢