算法Day12-层次遍历二叉树

题目

给定一个二叉树，返回其节点值的层次遍历（即从左到右，一层一层遍历）
例如：
给定二叉树{3,9,20,#,#,15,7}
3
/ \
9 20
/ \
15 7
返回层次遍历如下：
[
[3],
[9,20]
[15,7]
]

解析

通过广度优先遍历来实现层次遍历。创建一个Queue来缓存每一层的树节点，在遍历Queue的过程中，每取出一个元素，将该元素的左右子节点按顺序插入到Queue中。一直遍历下去，直到Queue为空。

代码

vector< vector<int> >levelOrder(TreeNode *root){
    vector< vector<int> > result;
    if(root == NULL) return result;
    queue<TreeNode*> nodeQ; //先进先出(FIFO) 队列类型的nodeQ变量用来缓存每一层的数节点
    nodeQ.push(root);   //push
    int nextLevelCnt=0, currentLeveCnt=1;
    vector<int> layer;  //layer存放的为某一层的节点数值，通过layer作为中间变量加入到result
    int visitedCnt=0;   //访问过的节点数目
    while(nodeQ.size() != 0)    //  队列不为空时访问，否则返回结果
    {
        TreeNode* node = nodeQ.front();
        nodeQ.pop();
        visitedCnt++;
        layer.push_back(node->val);
        if(node->left != NULL)      //为空时不做处理
        {
            //不为空则进队列
            nodeQ.push(node->left);
            nextLevelCnt++;
        }
        if(node->right != NULL)
        {
            nodeQ.push(node->right);
            nextLevelCnt++;
        }
        if(visitedCnt == currentLeveCnt)    //访问过的节点等于该层节点数时，开始下一层的访问。
        {
            //下一层访问开始，visitedCnt置0，当前层数节点数为上层的nextLevelCnt数，nextLevelCnt置0
            visitedCnt = 0;
            currentLeveCnt = nextLevelCnt;
            nextLevelCnt = 0;
            result.push_back(layer);    //将上层的节点加入结果
            layer.clear();
        }
    }

    return result;

}