算法通关村——二叉树的层序遍历

给你一个二叉树，请你返回其层序遍历从左到右的所有值（逐层的，从左到右访问所有结点）

 层层遍历，每一层都对应着下一层的结点。我们可以用队列来辅助遍历，将根节点放到队列中，然后根节点取出，它的值放到双数组里，它的左右子树继续放到队列中。

每一层的外部用一个size做判断，size==0的时候就是当前层结点遍历完成的时候

public class TreeNode {
    public static void main(String[] args) {
        Tree root = new Tree(3, null, null);
        int[] values = {1, 2, 4, 17, 6, 11, 9, 49, 31};
        root.initTree(root, values);

        ArrayList<List<Integer>> res = levelTree(root);
        for (int i = 0; i < res.size(); i++) {
            System.out.println(res.get(i));
        }
    }

    public static ArrayList<List<Integer>> levelTree(Tree root) {
        //层序遍历的结果
        ArrayList<List<Integer>> res = new ArrayList<>();
        //用来存放每一层的结点
        LinkedList<Tree> queue = new LinkedList<>();
        queue.add(root);
        while (queue.size() > 0) {
            //取出当层结点再加入它的子结点，实际的size会一直变化，所以在外部用变量把它保存固定住，方便每一层的遍历
            int size = queue.size();
            //每一层的结点元素的值都放到tmpList里
            ArrayList<Integer> tmpList = new ArrayList<>();
            for (int i = 0; i < size; i++) {
                //移出结点，它的值加入到当前层的tmpList里，同时判断当前结点下有没有左右子树，有的话加到队列中
                Tree remove = queue.remove();
                tmpList.add(remove.data);
                if (remove.leftTree != null) {
                    queue.add(remove.leftTree);
                }
                if (remove.rightTree != null) {
                    queue.add(remove.rightTree);
                }
            }
            //给结果加上一层
            res.add(tmpList);
        }
        return res;
    }
}

class Tree {
    int data;
    Tree leftTree;
    Tree rightTree;

    public Tree(int data, Tree left, Tree right) {
        this.data = data;
        this.leftTree = left;
        this.rightTree = right;
    }

    //树的初始化
    public Tree initTree(Tree root, int[] values) {
        for (int i = 1; i < values.length; i++) {
            insert(root, values[i]);
        }
        return root;
    }

    public void insert(Tree root, int data) {
        //比root大的元素放右子树，不然放左子树
        if (data > root.data) {
            if (root.rightTree == null) {
                root.rightTree = new Tree(data, null, null);
            } else {
                this.insert(root.rightTree, data);
            }
        } else {
            if (root.leftTree == null) {
                root.leftTree = new Tree(data, null, null);
            } else {
                this.insert(root.leftTree, data);
            }
        }
    }

}