剑指offer第32题 从上到下打印二叉树

    科技2024-01-27  118

    文章目录

    问题1:从上到下打印问题描述:解题思路:代码实现:算法复杂度分析: 问题2:一层一行打印问题描述:解题思路:代码实现:算法复杂度分析: 问题3:之字形打印二叉树问题描述:解题思路:代码实现:算法复杂度分析:

    问题1:从上到下打印

    问题描述:

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

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

    3 / \ 9 20 / \ 15 7

    返回:

    [3,9,20,15,7]

    解题思路:

      本题考查的是树的层次遍历,使用广度搜索(BFS)即可解决。创建一个队列,首先将根节点压入队列,然后循环取出队列中的节点进行遍历,将节点的值放入结果集中,并将该节点的左右子树分别压入队列中;需要注意的是压入队列的时候判断一下节点是否为空;防止出现空指针异常。   特殊值的处理,当传入的根节点为空时,返回一个空数组。

    代码实现:

    import java.util.ArrayList; import java.util.*; public class t32从上到下打印二叉树 { public int[] levelOrder(TreeNode root) { if(root==null) return new int[0]; ArrayList<Integer> res = new ArrayList<Integer>(); Queue<TreeNode> list = new LinkedList<TreeNode>(); //用于广度搜索的队列 list.add(root); while(!list.isEmpty()) { TreeNode temp = list.poll(); //取出队列中的元素 res.add(temp.val); if(temp.left!=null) //防止空指针异常 list.add(temp.left); //将左子树压入队列 if(temp.right!=null) list.add(temp.right); //将右子树压入队列 } //将list转换为数组 int[] arr = new int[res.size()]; int i = 0; for(Integer num: res) { arr[i++] = num; } return arr; } }

    算法复杂度分析:

    时间复杂度为:O(N);N 为二叉树的节点数量,即 BFS 需循环 N 次。空间复杂度为:O(N);最差情况下,即当树为平衡二叉树时,最多有 N/2 个树节点同时在 队列中,使用 O(N) 大小的额外空间。

    提交结果:


    问题2:一层一行打印

    问题描述:

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

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

    3 / \ 9 20 / \ 15 7

    返回其层次遍历结果:

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

    解题思路:

      这题的解题思路和上一题的差不多,也是用广度搜索(BFS),需要注意的是怎么样分层;其实在每一次进入循环的时候队列中剩余的节点就是上一层中的所有节点。所以在进入循环的时候我们获取队列的大小作为内部循环的次数即可。

    代码实现:

    import java.util.LinkedList; import java.util.List; import java.util.Queue; public class t32分层打印二叉树 { public List<List<Integer>> levelOrder(TreeNode root) { LinkedList<List<Integer>> res = new LinkedList<List<Integer>>(); Queue<TreeNode> queue = new LinkedList<TreeNode>(); if(root!=null) { queue.add(root); }else { return res; } while(!queue.isEmpty()) { List<Integer> temp = new LinkedList<Integer>(); //当前队列中剩余的节点就是上一层中的所有节点 for(int i=queue.size(); i>0; i--) { TreeNode t = queue.poll(); temp.add(t.val); if(t.left!=null) queue.add(t.left); if(t.right!=null) queue.add(t.right); } res.add(temp); } return res; } }

    算法复杂度分析:

    时间复杂度为:O(N);N 为二叉树的节点数量,即 BFS 需循环 N 次。空间复杂度为:O(N);最差情况下,即当树为平衡二叉树时,最多有 N/2 个树节点同时在 队列中,使用 O(N) 大小的额外空间。

    提交结果:


    问题3:之字形打印二叉树

    问题描述:

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

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

    3 / \ 9 20 / \ 15 7

    返回其层次遍历结果:

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

    解题思路:

      这题和上一题相比,需要注意的是存放的顺序,创建一个标志符用于控制每一行的存放顺序即可。这里需要注意的是不能通过节点加入队列的顺序去控制每一层打印的顺序,因为加入队列的顺序是层层影响的。   所以我们用存放进链表中的顺序来控制,一层从链表的头部开始添加元素,下一层就从链表的尾部开始添加元素。从而达到之字形打印的效果。

    错误的案例: 通过节点加入队列的顺序去控制每一层打印的顺序,如下:

    public List<List<Integer>> levelOrder(TreeNode root) { LinkedList<List<Integer>> res = new LinkedList<List<Integer>>(); Queue<TreeNode> queue = new LinkedList<TreeNode>(); boolean flag = false; if(root!=null) { queue.add(root); }else { return res; } while(!queue.isEmpty()) { List<Integer> temp = new LinkedList<Integer>(); //当前队列中剩余的节点就是上一层中的所有节点 for(int i=queue.size(); i>0; i--) { TreeNode t = queue.poll(); temp.add(t.val); if(flag) { if(t.left!=null) queue.add(t.left); if(t.right!=null) queue.add(t.right); }else { if(t.right!=null) queue.add(t.right); if(t.left!=null) queue.add(t.left); } } flag = !flag; //改变顺序 res.add(temp); } return res; }

    代码实现:

    import java.util.LinkedList; import java.util.List; import java.util.Queue; public class t32之字形分层打印二叉树 { public List<List<Integer>> levelOrder(TreeNode root) { LinkedList<List<Integer>> res = new LinkedList<List<Integer>>(); Queue<TreeNode> queue = new LinkedList<TreeNode>(); boolean flag = false; //用于控制存放的方向, if(root!=null) { queue.add(root); }else { return res; } while(!queue.isEmpty()) { LinkedList<Integer> temp = new LinkedList<Integer>(); //当前队列中剩余的节点就是上一层中的所有节点 for(int i=queue.size(); i>0; i--) { TreeNode t = queue.poll(); if(flag) { temp.addFirst(t.val); //添加在头部 }else { temp.add(t.val); //添加在尾部 } if(t.left!=null) queue.add(t.left); if(t.right!=null) queue.add(t.right); } flag = !flag; //改变顺序 res.add(temp); } return res; } }

    算法复杂度分析:

    时间复杂度为:O(N);N 为二叉树的节点数量,即 BFS 需循环 N 次。空间复杂度为:O(N);最差情况下,即当树为平衡二叉树时,最多有 N/2 个树节点同时在 队列中,使用 O(N) 大小的额外空间。和上一题相比,每一次多了一个条件判断。

    提交结果:

    Processed: 0.017, SQL: 9