给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
遍历二叉树一般用递归,如果是层次遍历一般用队列然后进行迭代,用递归也可以,因此这里使用两种方法,C++使用迭代,python使用递归。之所以使用队列是因为队列可以按层顺序来放置树节点指针,而且每次都可以取出队头来获取队头节点下一层的树节点,可以从开头想,如果开始放入根节点,那么进入循环,首先弹出的是根节点,然后由根节点再放入第二层的两个节点,然后依次访问第二层的两个节点,访问时又会放入第三层的四个节点,所以都是按层次顺序放入节点的,而且在访问某一层的同时放入下一层,所以这一层访问完之后,这一层也就都被弹出,而下一层也就都被放入,因此可以用队列来层次遍历二叉树。
然后说递归,递归有三个大要素:1. 递归结束的条件。2. 递归执行的内容。3. 执行递归。所以这里递归结束的条件肯定是遇到空节点。递归执行的内容就是把此次的节点的值保存。然后就是递归函数。主要的部分是执行内容,这里需要分层,每次执行递归的时候需要知道此时这个节点是哪一层的,递归是不会按层顺序去执行的,递归都是深度优先搜索策略,从左往右去搜索,所以必须标记此时节点的层。所以应该在递归之外设置二维数组,然后第一维表示层,第二维表示这一层的所有数。遍历的过程中就可以直接按层标记来记录所有数。
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) {
vector<vector<int>> res;
queue<TreeNode*> q;
if (root==NULL) return res;
q.push(root);
while (!q.empty())
{
vector<int> oneLevel;
int n = q.size();
for(int i=0;i<n;i++)
{
TreeNode* t = q.front();
q.pop();
oneLevel.push_back(t->val);
if (t->left!=NULL) q.push(t->left);
if (t->right!=NULL) q.push(t->right);
}
res.push_back(oneLevel);
}
return res;
}
};
python3源代码:(递归)
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def levelOrder(self, root):
"""
:type root: TreeNode
:rtype: List[List[int]]
"""
l = []
self.levelorder(root, 0, l)
return l
def levelorder(self, root, level, l):
if root==None:
return None
if len(l)==level:
l.append([])
l[level].append(root.val)
self.levelorder(root.left, level+1, l)
self.levelorder(root.right, level+1, l)