Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).
For example:
Given binary tree {3,9,20,#,#,15,7}
,
3 / \ 9 20 / \ 15 7
return its level order traversal as:
[ [3], [9,20], [15,7] ]
大意:层序遍历二叉树
解题思路:
1,因为是返回的是vector<vector<int>>,所以每一层遍历完毕后都要入大的vector,同时存储元素的临时vetcor要清空。
2,怎样来循环遍历呢?与其他遍历方式相同,总的循环条件仍然是队列不为空,但是技巧在于设定空指针。将root入队,同时null入队,循环中如果碰到非空节点,那么就将该节点左右子树分别入队,直至空节点。碰到空节点,就将刚才的临时vector存储起来,再清空,同时队列中再加入一个null,来区分时候一层完毕。
代码部分:
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector
> levelOrder(TreeNode *root) {
vector
> vec;
if(root==NULL)
{return vec;}
vector
lev; queue
que; que.push(root); que.push(NULL); while(!que.empty()) { TreeNode* temp=que.front(); que.pop(); if(temp!=NULL) { lev.push_back(temp->val); if(temp->left) que.push(temp->left); if(temp->right) que.push(temp->right); } else { vec.push_back(lev); if(!que.empty()) { lev.clear(); que.push(NULL); } } } return vec; } };
图示:
代码解析:
1,当遍历到空指针的时候,说明应该将刚才入vector的指针节点的集合 入最终结果的vector,但是必须要判断结束条件。也就是说只要接下来还有层的时候才能继续遍历和存储。
2,最后依次更新4,5,6的时候并没有继续往指针队列中添加指针,因为他们的左右子树都为空,所以,他们弹出之后,整个指针队列已经没有指针了,如果不加限制队列为空的条件,那么将一直循环下去,每次都会往结果里面加空集合。