打印二叉树

de不过bug

于 2020-03-20 16:45:21 发布

阅读量211

点赞数

分类专栏：刷刷题

本文链接：https://blog.csdn.net/fengqidong93/article/details/104992793

版权

刷刷题专栏收录该内容

2 篇文章 0 订阅

订阅专栏

本文介绍了如何根据不同的需求层次遍历二叉树：从上到下按左到右顺序，同一层按行打印，以及之字形顺序打印。通过广度优先搜索策略，使用队列或列表实现这些遍历方法。

摘要由CSDN通过智能技术生成

剑指offer第32题,打印二叉树,一共三小问

1.从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。

例如:
给定二叉树: [3,9,20,null,null,15,7],

[3,9,20,15,7]

其实看到这个问题的时候,就明白这是个广度优先遍历的问题,二叉树这种结构,并没有规范的遍历条件,所以面对遍历二叉树的问题,使用递归遍历的偏多.如果一定要求遍历呢?我们有什么数据结构可以利用?我们希望的是一层一层的遍历完这棵树,如何保障全部都遍历到呢?我想到的是找一个容器,从根节点开始遍历,并将子节点放入容器

//我使用了队列来完成,如果不熟悉队列,用list也可以完成这一操作,只是需要记录list的索引,确保自己都遍历到

class Solution {
    public int[] levelOrder(TreeNode root) {
        if(root==null)
            return new int[0];
        Queue<TreeNode> queue = new LinkedList<TreeNode>();//初始化一个队列用来存放树的节点
        queue.offer(root);
        ArrayList ans =new ArrayList<Integer>();
        while(!queue.isEmpty()){
            TreeNode node = queue.poll();
            
            ans.add(node.val);
            if (node.left != null )
                queue.offer(node.left);
            if (node.right != null)
                queue.offer(node.right);
        }
        int [] res = new int[ans.size()];
        for(int i =0;i<ans.size();i++){
            res[i]=(int)ans.get(i);
        }
        return res;

    }
}

2.从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

例如:
给定二叉树: [3,9,20,null,null,15,7],

返回其层次遍历结果：

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

思考这个问题,发现他与第一问的区别是希望把每一层分开打印.结合之前的代码,可以看出,我们是在遍历一层的时候,向容器加入他的下一层,那么如果为每一层创建一个list即可完成.

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> result =new ArrayList();
        if(root==null)
            return result;
        
        Queue<TreeNode> queue = new LinkedList();
        queue.offer(root);
        while(!queue.isEmpty()){
            int size = queue.size();//当前队列的长度,就是该层元素的个数
            ArrayList<Integer> list=new ArrayList();
            for(int i=0;i<size;i++){
                TreeNode node= queue.poll();
                list.add(node.val);//依次遍历该层的元素,并将它们的孩子节点放入队列
                if(node.left!=null)
                    queue.offer(node.left);
                if(node.right!=null)
                    queue.offer(node.right);
            }
            result.add(list);
        }
        return result;
    }
}

3.请实现一个函数按照之字形顺序打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右到左的顺序打印，第三行再按照从左到右的顺序打印，其他行以此类推。

例如:
给定二叉树: [3,9,20,null,null,15,7],

返回其层次遍历结果：

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

如果我们能够完成第二问,第三问就比较轻松了.直接遍历第二问的答案,隔行进行reverse即可

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> result =new ArrayList();
        if(root==null)
            return result;
        
        Queue<TreeNode> queue = new LinkedList();
        queue.offer(root);
        int num =0;
        while(!queue.isEmpty()){
            int size = queue.size();
            ArrayList<Integer> list=new ArrayList();
            for(int i=0;i<size;i++){
                TreeNode node= queue.poll();
                list.add(node.val);
                if(node.left!=null)
                    queue.offer(node.left);
                if(node.right!=null)
                    queue.offer(node.right);
            }
            if(num%2==0){
                result.add(list);
            }else{
                Collections.reverse(list);
                result.add(list);
            }
            num++;
        }
        
        return result;
    }
}

很基础的题目,也是很基础的题解.有什么疑问欢迎留言