//2.二叉树的分层遍历(逐层的返回其按照层序遍历得到的节点值,从左到右访问所有节点)
static List<List<Integer>> result = new ArrayList<>();
public List<List<Integer>> levelOrder(TreeNode root) {
//此处要把每一层的节点放到一个单独的List中,所以之前的层序遍历方式就不行
//此处要基于递归解决问题
if (root == null) {
return result;
}
//helper方法辅助递归,第二个参数表示当前的层数(层数从0开始,和list下标对应)
helper(root,0);
return result;
}
private void helper(TreeNode root, int lever) {
if (lever == result.size()) {
result.add(new ArrayList<>());
}
//把当前节点添加到result中的合适位置
result.get(lever).add(root.val);
//先根据lever取到result中的第几行,再把root.val这个元素加到这一行。
if (root.left != null) {
helper(root.left,lever+1);
}
if (root.right != null) {
helper(root.right,lever+1);
}
}