////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
例如:
给定二叉树: [3,9,20,null,null,15,7]
,
3 / \ 9 20 / \ 15 7
返回其层次遍历结果:
[ [3], [9,20], [15,7] ]
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
/*
这道题用递归做是可以的,但是跟前面的几道树的题有点区别的是,这道题并不是“判断”,比如在二叉树的最大深度中,只要判断是否到达叶子节点就能得出深度,将几个深度比较就能得到最大深度。
所以就有一个问题了,递归的话必须要有一个“出口”,不然一直递归下去没意思啊,之前都是用返回值的形式跳出递归,在这道题目中当然也可以这么做(我就是这么做),不过返回一个没什么意思的返回值总感觉有点膈应人。
*/
给定的函数返回值类型是二维数组(二维vector??),那思路就很简单了,递归到每一层都把值(如果有的话)push_back到 一个vector(每一层只有一个,如何做到后面介绍),然后所有的vector再push_back到要返回的二维vector中(当然真正操作的时候是反着的,先把每一层的vector push_back到返回的二维vector中,然后再填充每一层的vector)。
首先确定递归函数的参数。递归函数的任务是把节点的值push_back到每层的vector中,并且在需要的时候把层的vector push_back到二维vector中(再次强调具体操作时的顺序是反着的,只是这样说比较容易理解),所以很明显要有一个节点类型的指针用于递归遍历,还要有一个int类型的值指示层数,以及一个存储结果的二维vector。
bool pushBack(int level, TreeNode *node, vector<vector<int>> & res){}
要执行递归函数也很简单,直接调用就是了,虽然有返回,但是在题目中不需要,所以并不需要变量接收返回值。
函数的基本逻辑是遇到叶子节点就返回,即判断传入的node是不是空。如果不是的话继续判断二维vector res的大小是不是小于层数level+1,如果是的话,说明二维数组还没有接收到树在这层的数据,需要创建一个vector,并把它push_back进去。接下来只要把节点的值push_back到res的第level层中,然后leve++,递归遍历左右子树即可。
贴上完整的代码。
//4ms
class Solution
{
public:
bool pushBack(int level, TreeNode *node, vector<vector<int>> & res)
{
int l;
if (node == NULL)
return true;
if (res.size() < level + 1)
{
vector<int> v;
res.push_back(v);
}
res[level].push_back(node->val);
level++;
pushBack(level, node->left, res);
pushBack(level, node->right, res);
return true;
}
vector<vector<int>> levelOrder(TreeNode* root)
{
vector<vector<int>> res;
if (root == NULL)
return res;
pushBack(0, root, res);
return res;
}
};