题目描述
题目链接:https://leetcode-cn.com/problems/binary-tree-level-order-traversal/
给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
解题方案
- 迭代
设置一个队列,对二叉树进行层次遍历,遍历到的节点压入队列;弹出某个节点后将其左右孩子节点依次压入队列中。
设置两个变量,记为last,nlast,其中last用于记录当前行最右节点,nlast记录下一行最右节点(nlast始终记录最新进入队列的节点)。当last等于最新打印的节点,则是last=nlast,这样就完成了换行操作。
class Solution {
public:
vector< vector<int> > levelOrder(TreeNode* root) {
vector<vector<int>> res;
if(!root)
return res;
vector<TreeNode *> level;
level.push_back(root);
while(!level.empty())
{
vector<TreeNode *> temp;
vector<int> level_val;
for(int i=0;i<level.size();++i)
{
if(level[i]->left)
temp.push_back(level[i]->left);
if(level[i]->right)
temp.push_back(level[i]->right);
level_val.push_back(level[i]->val);
}
res.push_back(level_val);
level = temp;
}
return res;
}
};
执行用时16ms,内存消耗14MB。
- 递归
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> res;
preorder(root, 0, res);
return res;
}
void preorder(TreeNode *root, int depth, vector<vector<int>> &res)
{
if(!root)
return;
if(depth>=res.size())
res.push_back(vector<int> {});
res[depth].push_back(root->val);
preorder(root->left, depth+1, res);
preorder(root->right, depth+1, res);
}
};
执行用时16ms,内存消耗14.7MB。