给定一个二叉树,返回其节点值自底向上的层次遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)
例如:
给定二叉树 [3,9,20,null,null,15,7]
,
3 / \ 9 20 / \ 15 7
返回其自底向上的层次遍历为:
[ [15,7], [9,20], [3] ]
这道题主要是记录discuss的超简洁代码解法。
我的第一想法是用两个queue,第一个queue存父节点,每次第一个queue所有元素出队列,对应的子节点进入第二个queue队列,然后第二个queue所有元素出队列,对应的子节点入第一个queue,这样不断循环利用两个queue可以实现树的层序遍历,然后每层的节点保存到一个stack中,最后stack出栈便是答案,想想都觉得代码很多。。。然后看了discuss部分,有个大神的代码及其简介,这里对其做介绍并做相应记录。
首先得到树的深度d,然后直接从最深层(叶子结点层)开始为vactor<vactor<int>> res赋值,然后传入左右子节点递归调用自身,代码不要太简洁!!!
代码有详细注释:
//得到树深度
int depth(TreeNode* root) {
if (!root) {
return 0;
}
return max(depth(root->left), depth(root->right)) + 1;
}
//这里直接给res赋值,然后递归调用左右节点
void levelOrderBottom(TreeNode* root, int depth, vector<vector<int>>& res) {
if (root) {
res[depth].push_back(root->val);
levelOrderBottom(root->left, depth - 1, res);
levelOrderBottom(root->right, depth - 1, res);
}
}
vector<vector<int>> levelOrderBottom(TreeNode* root) {
int d = depth(root);
vector<vector<int>> res = { d,vector<int>{} };
levelOrderBottom(root, d - 1, res);
return res;
}