给定一个二叉树,返回其节点值自底向上的层次遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其自底向上的层次遍历为:
[
[15,7],
[9,20],
[3]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-level-order-traversal-ii
方法1:队列和map,
一开始想的是使用map,key存放level,value存放对应的值。但是发现几乎没有人这样做,大家都采用的是栈和队列。不过也算是复习了一下map。
思想与leetcode102很像,只不过在输出方式上发生变化。
class Solution {
public:
vector<vector<int>> levelOrderBottom(TreeNode* root) {
unordered_map<int,vector<int>> res;
if(!root) return {};
queue<pair<TreeNode*, int>> q;
q.push(make_pair(root, 0));
while(!q.empty()){
TreeNode* node = q.front().first;
int level = q.front().second;
q.pop();
res[level].push_back(node->val); //将level层的节点值放入。
if(node->left) q.push(make_pair(node->left, level + 1));
if(node->right) q.push(make_pair(node->right, level + 1));
}
int n = res.size();
vector<vector<int>> result;
for(int i = n-1; i >= 0; i--)
result.push_back(res[i]); // 将level从大到小放入
return result;
}
};
方法2:
使用栈和队列。
class Solution {
public:
vector<vector<int>> levelOrderBottom(TreeNode* root) {
if(root == NULL) return {};
vector<vector<int>> res;
queue<TreeNode*> q; // 产生层次遍历
stack<vector<int>> s; //保存层次遍历的结果
q.push(root);
while(!q.empty()){
int len = q.size(); // 这一层的节点数
vector<int> temp;
for(int i = 0; i < len; i++){
TreeNode* node = q.front();
q.pop();
temp.push_back(node->val);
if(node->left) q.push(node->left);
if(node->right) q.push(node->right);
}
s.push(temp);
}
while(!s.empty()){
res.push_back(s.top());
s.pop();
}
return res;
}
};