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