算法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;}