LeetCode102题---二叉树的层序遍历

LeetCode102题—二叉树的层序遍历
给你一个二叉树，请你返回其按 层序遍历 得到的节点值。 （即逐层地，从左到右访问所有节点）。

迭代实现：

广度优先遍历是按层层推进的方式，遍历每一层的节点。题目要求的是返回每一层的节点值，所以这题用广度优先来做非常合适。
广度优先需要用队列作为辅助结构，我们先将根节点放到队列中，然后不断遍历队列。

拿出根节点，如果左子树/右子树不为空，就将他们放入队列中。
第二次处理，会将 2 和 5 这两个节点从队列中拿走，然后再将 2 和 5 的子节点放入队列中，现在队列中就有三个节点 3，4，6。

public List<List<Integer>> levelOrder(TreeNode root) {
        if (root == null) { //判断是否为空树
            return new ArrayList<List<Integer>>();
        }

        List<List<Integer>> list = new ArrayList<>();
        LinkedList<TreeNode> queue = new LinkedList<>();
        //将根节点放到队列中,然后不断遍历队列
        queue.add(root);
        while (queue.size() > 0) {
            //获取当前队列的长度。这个长度相当于当前层的元素的个数
            int length = queue.size();
            ArrayList<Integer> level = new ArrayList<>();//创建临时的一个list
            //将队列中的元素取出来放到临时list中
            //如果节点的左子节点、右子节点都不为空，也放到队列中
            for (int i = 0; i < length; i++) {
                TreeNode node = queue.remove();
                level.add(node.val);
                if (node.left != null) {
                    queue.add(node.left);
                }
                if (node.right != null) {
                    queue.add(node.right);
                }
            }
            list.add(level);
        }
        return list;
    }

递归实现

按照深度优先的处理顺序，会先访问节点 1，再访问节点 2，接着是节点 3。
之后是第二列的 4 和 5，最后是第三列的 6。

每次递归的时候都需要带一个 index(表示当前的层数)，也就对应那个田字格子中的第几行，如果当前行对应的 list 不存在，就加入一个空 list 进去。

 public List<List<Integer>> levelOrder(TreeNode root) {
        if(root == null){ //判断是否为空树
            return new ArrayList<List<Integer>>();
        }
        List<List<Integer>> list = new ArrayList<>();
        dfs(root,list,1);//从第一层开始
        return list;
    }
    /*参数： 节点  存放元素的链表 层数 */
    public void dfs(TreeNode root,List<List<Integer>> list,int index){
        //如果不是空树，就是从第一层开始的。初始化的list大小为0，是要比层数小的
        //假设res是[ [1],[2,3] ]， index是3，就再插入一个空level放到list中
        if(list.size() < index){
            list.add(new ArrayList<Integer>());
        }

        //将当前节点的值加到list中，index表示当前层， 将当前层的的level取出来，将当前节点的值存到这个level中
        list.get(index - 1).add(root.val);

        //递归处理左子树，右子树，层数也加一
        if(root.left != null){
            dfs(root.left,list,index+1);
        }
        if(root.right != null){
            dfs(root.right,list,index+1);
        }

    }