剑指offer-把二叉树打印成多行

题目描述

从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。

思路与实现

way1 BFS

可以看出符合先进先出的特点，所以用队列实现。
results为结果列表
temp记录某层的临时列表
queue为待处理的结点队列

1. 首先将根结点加入待处理的结点队列，当带处理队列不为空，queue.size()获得该层的所有结点的个数。

2. 依次取出该层的结点，并从队列中删除；每次处理一个，将的值val加入记录某层的临时列表temp;如果待处理结点有左右孩子结点，则将左右孩子结点加入到队列中。

3. 本层处理完，深层拷贝将本层结果（存放在临时列表tmp）加入结果列表results。

4. 当处理的结点队列q为空，所有结点处理完，结束。

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;


/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> results = new ArrayList<>();
        if (pRoot == null)
            return results;
        //1.创建一个队列，把起始节点放到里面
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(pRoot);
        // 2.while队列不为空，处理队列中的节点，并拓展出新的结点
        while (!queue.isEmpty()) {
            ArrayList<Integer> temp = new ArrayList<>();
            //获得当前层结点的个数
            int size = queue.size();
            //for 上一层的节点 拓展到下一层的节点
            for (int i = 0; i < size; i++) {
                TreeNode node = queue.poll();
                temp.add(node.val);
                if (node.left != null)
                    queue.offer(node.left);
                if (node.right != null)
                    queue.offer(node.right);
            }
            results.add(new ArrayList<>(temp));//深层拷贝

        }
        return results;
    }
}

way2 递归

//用递归做
import java.util.ArrayList;
import java.util.Stack;
public class Solution {
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> list = new ArrayList<>();
        depth(pRoot, 1, list);
        return list;
    }
     
    private void depth(TreeNode root, int depth, ArrayList<ArrayList<Integer>> list) {
        if(root == null) 
            return;
        if(depth > list.size())
            list.add(new ArrayList<Integer>());
        list.get(depth -1).add(root.val);
        depth(root.left, depth + 1, list);
        depth(root.right, depth + 1, list);
    }
}