给出一棵二叉树,返回其节点值的层次遍历(逐层从左往右访问)
样例
给一棵二叉树 {3,9,20,#,#,15,7}
:
3
/ \
9 20
/ \
15 7
返回他的分层遍历结果:
[
[3],
[9,20],
[15,7]
]
代码:
class Solution {
std::vector<vector<int>> v2;
std::vector<int> v1;
queue<TreeNode*> q;
TreeNode *top;
int l;
/**
* @param root: The root of binary tree.
* @return: Level order a list of lists of integer
*/
public:
vector<vector<int>> levelOrder(TreeNode *root) {
// write your code here
if(root==NULL)
return v2;
q.push(root);
while(!q.empty())
{
l=q.size();
while(l!=0)
{
top=q.front();
v1.push_back(top->val);
l--;
if(top->left!=NULL)
q.push(top->left);
if(top->right!=NULL)
q.push(top->right);
q.pop();
}
v2.push_back(v1);
v1.clear();
}
return v2;
}
};
感想:这题实在借鉴了别人的代码后写出的,思路是返回值是一个以向量为元素的向量,先在函数外定义两个向量std::vector<vector<int>> v2;和std::vector<int> v1;将一层的节点的数值放入v1,完成一个v1的放入就将v1为一个元素放入v2中,清空v1,便于下一层的节点的数进入v1中。建立一个队列暂存一层的结点,符合先进先出的原则。先将跟节点放入队列中,记录队列的节点的个数。最开始将根节点放入队列,长度为q.size(),并且循环这些次,保证了一层的节点的值进入同一个v1。将节点的数值放入v1中,如果当前节点的左节点不是空就放入队列,右节点相同,按q.soze()循环完放入v2,同时清空v1。当队列中再无节点时结束程序,反回v2。