给定一个二叉树,返回其节点值自底向上的层序遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)。
示例 1:
Input: root = [3,9,20,null,null,15,7]
Output: [[15,7],[9,20],[3]]
示例 2:
Input: root = [1]
Output: [[1]]
示例 3:
Input: root = []
Output: []
提示:
- -1000 ≤
Node.val
≤ 1000 - 树中的节点数在
[0,2000]
范围内
思路
本题在102题:二叉树的层序遍历基础上修改。前几步和原来一样,到了最后一步,我们把结果从屁股到脑袋抄到一个新数组里,用空间换时间。或者调用reverse()
函数对原来的结果进行反转。
代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<vector<int>> levelOrderBottom(TreeNode* root) {
vector< vector<int> > ans;
if( root == nullptr ) return ans;
if( root->left == nullptr && root->right == nullptr ) {
vector<int> tmp(1, root->val);
ans.emplace_back(tmp);
return ans;
}
queue<TreeNode*> Q;
Q.push(root);
while( !Q.empty() ) {
int walk_length = Q.size();
vector<int> layer;
for(int i = 0; i < walk_length; i++) {
TreeNode* cur = Q.front();
Q.pop();
layer.emplace_back( cur->val );
if( cur->left != nullptr ) Q.push( cur->left );
if( cur->right != nullptr ) Q.push( cur->right );
}
ans.emplace_back( layer );
}
// Copy to a new vector in reverse order
vector< vector<int> > res;
for(int i = ans.size()-1; i >= 0; i--) {
res.emplace_back( ans[i] );
}
return res;
}
};