题目描述:给出一棵二叉树,返回其节点值的层次遍历(逐层从左往右访问)
样例:
给一棵二叉树 {3,9,20,#,#,15,7}
:
3
/ \
9 20
/ \
15 7
返回他的分层遍历结果:
[
[3],
[9,20],
[15,7]
]
做题思路:创建队列,把根节点放入队列,建立一个length用来记录队列长度,把树的结点指针指向队列中的第一个数,取值放入数组1中,把队列的数弹出,然后长度随之减少,根节点的左孩子如果不是空,就放入队列,右孩子不为空则放入队列,放入数组后弹出,清空数组,再继续循环,数组1放入数组2中,直到队列为空。
关键代码:
class Solution {
/**
* @param root: The root of binary tree.
* @return: Level order a list of lists of integer
*/
public:
std::vector<int> f;
std::vector<vector<int>> w;
vector<vector<int>> levelOrder(TreeNode *root) {
// write your code here
if(root==NULL) return w;
queue<TreeNode *> q;
q.push(root);
int length;
while(!q.empty())
{ length=q.size();
while(length!=0)
{ TreeNode *t=q.front();
f.push_back(t->val);
q.pop();
length--;
if(t->left!=NULL)
q.push(t->left);
if(t->right!=NULL)
q.push(t->right);
}
w.push_back(f);
f.clear();
}
return w;
}
};
做题感想:刚开始自己不会写,后来上网查了查,然后自己按照理解重新写了,前几遍输出的答案和期望答案不一样,后来给同学讲了一遍,讲的过程中突然发现,小vector一直没有清空,加了clear的语句之后就对了。