LeetCode 102 二叉树的层序遍历

题目

给你一个二叉树，请你返回其按 层序遍历 得到的节点值。 （即逐层地，从左到右访问所有节点）。
示例：
二叉树：[3,9,20,null,null,15,7],

3
/
9 20
/
15 7
返回其层次遍历结果：

[
[3],
[9,20],
[15,7]
]

算法思路，BFS

层序遍历，关于BFS广度优先搜索。使用队列来实现。
输出时，还需要分层，一层一层的输出。所以我们还需要一个标记，来标记队列里那些的同一层的。

直接看代码，有注释

代码

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) { 
        if(root == null)
             return new ArrayList<>();  //返回一个空的数组
        List<List<Integer>> res = new ArrayList<>();  //放层数 （每一层里面放元素）
        Queue<TreeNode>  queue = new LinkedList<>();
        queue.add(root);
        while(!queue.isEmpty()){  //队列不空
            int a = queue.size();   //记住现在队列里的个数，就是同一层元素的个数
            List<Integer> list = new ArrayList<>();  //放同一层的元素
            while(a > 0){
                TreeNode node = queue.poll();  
                list.add(node.val);  //同一层元素放到数组list
                if(node.left != null)
                    queue.add(node.left);
                if(node.right != null)
                    queue.add(node.right);
                a--;  //  直到同一层元素全部移除
            }
            res.add(list); //将放同一层的元素的数组list ，放入到放层数的数组
        }     
        return res;
    }
}

思路二 ，使用DFS，

使用DFS的话，因为每一次的深度优先搜索，会记住当前的层次。然后把相同层次的数值 ，放在同一层中，在放入结果就好了

class Solution {
	public List<List<Integer>> levelOrder(TreeNode root) {
		if(root == ull) {
			return new ArrayList<List<Integer>>();
		}
		List<List<Integer>> res = new ArrayList<List<Integer>>();
		dfs(1,root,res); //根结点为第一层
		return res;
	}
	
	public void dfs(int index,TreeNode root, List<List<Integer>> res) {
		//res中的size 就是一共会有的层数，如果层数没到，就创建
		if(res.size() < index) {
			res.add(new ArrayList<Integer>());
		}
		//将当前节点的值加入到res中，index代表当前层，假设index是3，节点值是12
		//res是[ [1],[2,3] [4] ]，加入后res就变为 [ [1],[2,3] [4,12] ]
		//因为递归是 左子树到右子树，所以res中list的顺序 也是从左到右
		res.get(index-1).add(root.val);
		//递归的处理左子树，右子树，同时将层数index+1
		if(root.left!=null) {
			dfs(index+1, root.left, res);
		}
		if(root.right!=null) {
			dfs(index+1, root.right, res);
		}
	}
}