2021-09-11

从上到下打印二叉树

 class Solution {
        /*
         * 题目：从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。
         *
         * 思路：广搜
         *
         * */
        public int[] levelOrder(TreeNode root) {
            if (root == null) {
                return new int[0];
            }
            Queue<TreeNode> queue = new LinkedList<>();
            queue.add(root);
            List<Integer> res = new LinkedList<>();
            while (!queue.isEmpty()) {
                int n = queue.size();
                for (int i = 0; i < n; i++) {
                    TreeNode node = queue.poll();
                    res.add(node.val);
                    if (node.left != null) {
                        queue.add(node.left);
                    }
                    if (node.right != null) {
                        queue.add(node.right);
                    }
                }
            }
            int[] nums = new int[res.size()];
            int index = 0;
            for (int t : res) {
                nums[index++] = t;
            }
            return nums;
        }
    }

从上到下打印二叉树 II

 

class Solution {
        /*
        *
        * 题目：从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。
        *
        * 思路：广搜
        *
        * */
        public List<List<Integer>> levelOrder(TreeNode root) {
            if (root == null) {
                return new LinkedList<>();
            }
            Queue<TreeNode> queue = new LinkedList<>();
            queue.add(root);
            List<List<Integer>> res = new LinkedList<>();
            while (!queue.isEmpty()) {
                int n = queue.size();
                List<Integer> list = new LinkedList<>();
                for (int i = 0; i < n; i++) {
                    TreeNode node = queue.poll();
                    list.add(node.val);
                    if (node.left != null) {
                        queue.add(node.left);
                    }
                    if (node.right != null) {
                        queue.add(node.right);
                    }
                }
                res.add(list);
            }
            return res;
        }
    }

 

class Solution {
        /*
         *
         * 题目：请实现一个函数按照之字形顺序打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右到左的顺序打印，第三行再按照从左到右的顺序打印，其他行以此类推。
         *
         * 思路：广搜的变种题，使用链表来记录每一层，然后输出
         *
         * 知识点：LinkedList.addFirst     List 没有这个函数
         *
         * */
        public List<List<Integer>> levelOrder(TreeNode root) {
            if (root == null) {
                return new LinkedList<>();
            }
            int row = 0;
            List<List<Integer>> res = new LinkedList<>();
            List<TreeNode> rowList = new LinkedList<>();
            rowList.add(root);
            while (!rowList.isEmpty()) {
                LinkedList<Integer> numList = new LinkedList<>();
                List<TreeNode> nodeList = new LinkedList<>();
                while (!rowList.isEmpty()) {
                    TreeNode node = rowList.remove(0);
                    if (row % 2 == 0) {
                        numList.add(node.val);
                    }else {
                        numList.addFirst(node.val);
                    }
                    if(node.left != null){
                        nodeList.add(node.left);
                    }
                    if(node.right != null){
                        nodeList.add(node.right);
                    }
                }
                rowList = nodeList;
                row ++;
                res.add(numList);
            }
            return res;
        }
    }