题目要求
给定一个二叉树,从底到顶的返回节点值的层序遍历结果。
(tips:从左到右,一层一层)
注意和解答:leetcode102 Binary Tree Level Order Traversal(二叉树的层序遍历)的不同。
输入示例
For example:
给定一个二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
return 从底到顶的层序遍历结果如下:
[
[15,7],
[9,20],
[3]
]
解题思路
根据输入示例的结果形式可知,我们需要两个vector来辅助解答。
一个vector存放每一层的遍历结果,一个vector存放最后的整体结果。
根据层序遍历的要求(从左到右,一层一层的特点),我们选用队列来实现(先进先出),试想一下,每次我们取队首元素,然后把它的值放入当前层的vector中,再把它的孩子节点放到队列里进行迭代,只要队列中还有元素我们就进行循环操作,直到到达叶子结点。
按照上述步骤,我们是根据从上到下即从顶到底的顺序进行遍历的,根据题目要求,我们最后把vector中的元素倒叙即可。
, 难点是:什么时候我们结束当前一层遍历去进行下一层呢?
答案:这里通过当前层中-队列-里元素的个数就能知道当前层进行几次循环取值操作,用for来进行循环取值。比如第二层有9,20两个元素,要进行两个for循环(注意这里有个大坑点,不要在for循环中使用q.size()因为这个队列q在for循环中是不断改变的)
主要代码(c++)
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
if(root==NULL) return {};
vector<vector<int>>res; // 第一个vector
queue<TreeNode*>q;
q.push(root);
while(!q.empty())// 迭代的条件
{
int size = q.size(); //这里要先保存下来队列的元素个数,不然在for中会改变
vector<int>currentlevel; //第二个vector
for(int i =0; i<size; ++i) //大坑点! 你可以把size换成q.size()试试看
{
TreeNode * tmp = q.front(); // 下面是常规写法 记录每一层的值
q.pop();
currentlevel.push_back(tmp->val);
if(tmp->left !=NULL) q.push(tmp->left);
if(tmp->right !=NULL) q.push(tmp->right);
}
res.push_back(currentlevel); // 或者不进行倒序操作,把这里变成
//res.insert(res.begin(),currentlevel)
//每次把层序结果插到最前边
}
reverse(res.begin(),res.end());
return res;
}
};
主要代码python
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def levelOrderBottom(self, root):
"""
:type root: TreeNode
:rtype: List[List[int]]
"""
res = []
if not root:
return res
queue = []
queue.append(root)
while queue:
eachlevel = []
size = len(queue)
for _ in range(size):
tmp = queue.pop(0)
eachlevel.append(tmp.val)
if tmp.left:
queue.append(tmp.left)
if tmp.right:
queue.append(tmp.right)
res.insert(0,eachlevel)
return res
划重点:
有关层序遍历的迭代记得用 队列哈!!!
相似题目:
解答:leetcode102 Binary Tree Level Order Traversal(二叉树的层序遍历)