leecode 102. 二叉树的层序遍历

102. 二叉树的层序遍历

本题仍然有两种做法，递归和迭代，详细见代码。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */

 迭代：

迭代方法借助Queue创建队列先进先出，与前序遍历的迭代方法和代码类似，但是DFS的遍历用栈结构
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        Queue<TreeNode> que = new LinkedList<>();
        List<List<Integer>> resList = new ArrayList<List<Integer>>();
        if (root == null) return resList;
        que.offer(root);
        while (!que.isEmpty()) {
//创建一个List放遍历后的节点值，len判断节点的数量
            List<Integer> itemList = new ArrayList<Integer>();
            int len = que.size();
//每遍历一次就让len--，这样当节点数量==0时说明遍历结束，大于0时说明这一层需要继续遍历
            while(len > 0) {
//和DFS的迭代法实现代码几乎一样
                TreeNode node = que.poll();
                itemList.add(node.val);
                if (node.left != null) que.offer(node.left);
                if (node.right != null) que.offer(node.right);
                len--;
            }
//最后需要向正式结果加入这一层的遍历结果，然后返回继续循环遍历下一层。
            resList.add(itemList);
        }
        return resList;
    }
}

递归：

 //DFS--递归方式
        class Solution {
        public List<List<Integer>> resList = new ArrayList<List<Integer>>();

        public List<List<Integer>> levelOrder(TreeNode root) {
            //checkFun01(root,0);
            checkFun02(root);

            return resList;
        }
       public void checkFun01(TreeNode node, Integer deep) {
        if (node == null) return;
        deep++;

        if (resList.size() < deep) {
            //当层级增加时，list的Item也增加，利用list的索引值进行层级界定
            List<Integer> item = new ArrayList<Integer>();
            resList.add(item);
        }
        resList.get(deep - 1).add(node.val);

        checkFun01(node.left, deep);
        checkFun01(node.right, deep);
    }