算法练习Day15| 二叉树的层序遍历 107. 二叉树的层序遍历 II 199. 二叉树的右视图 637. 二叉树的层平均值

#算法#leetcode#二叉树

今天主要练习二叉树的层序遍历衍生出来的几道题，我看难度等级都是中等，但是理解了层序遍历之后，其实这几道题思考下都是可以解决的，只是一些细节的变化。

再记录一下层序遍历的思路：

1. 创建队列，因为我们每一层要从左到右依次遍历，所以是一个先进先出的顺序；

2. 头结点入队，这里省去了对头结点的校验，一般都是要做空节点校验。

3. 遍历队列，记录每一层的节点个数，从队列里面去对应个节点，保存到结果。

107. 二叉树的层序遍历 II

题目链接：107. 二叉树的层序遍历 II - 力扣（LeetCode）给你二叉树的根节点 root ，返回其节点值 自底向上的层序遍历 。 （即按从叶子节点所在层到根节点所在的层，逐层从左向右遍历）

这道题思路就是先从上到下的层序遍历，然后再反转结果List即可。

具体代码：

public List<List<Integer>> levelOrderBottom(TreeNode root) {
        //从上到下遍历，然后反转
        List<List<Integer>> res = new ArrayList<List<Integer>>();
        if(null == root) {
            return  res;
        }
        
        Queue<TreeNode> queue = new ArrayDeque<>();
        queue.add(root);
        while(!queue.isEmpty()) {
            List<Integer> list = new ArrayList<>();
            int size = queue.size();
            while(size > 0) {
                TreeNode node = queue.poll();
                list.add(node.val);
                if(node.left != null) {
                    queue.add(node.left);
                }
                if(node.right != null) {
                    queue.add(node.right);
                }
                size--;
            }
            res.add(list);
        }

        Collections.reverse(res);
        return res;
    }

199. 二叉树的右视图

题目链接：199. 二叉树的右视图 - 力扣（LeetCode）给定一个二叉树的 根节点 root，想象自己站在它的右侧，按照从顶部到底部的顺序，返回从右侧所能看到的节点值。

这道题就是记录每一层最后一个节点到List中，因为我们是从左到右遍历的，所以取最后一个节点。

具体代码：

public List<Integer> rightSideView(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        if(null == root) {
            return res;
        }
        Queue<TreeNode> queue = new ArrayDeque<>();
        queue.add(root);
        while(!queue.isEmpty()) {
            int size = queue.size();
            while(size > 0) {
                TreeNode node = queue.poll();
                if(1 == size) {
                    res.add(node.val);
                }
                if(null != node.left) {
                    queue.add(node.left);
                }
                if(null != node.right) {
                    queue.add(node.right);
                }
                size--;
            }
        }
        return res;

    }

可以看到，当每一层的size == 1时，此时就是最后一个节点，保存到res列表中即可。

637. 二叉树的层平均值

题目链接：637. 二叉树的层平均值 - 力扣（LeetCode）给定一个非空二叉树的根节点 root , 以数组的形式返回每一层节点的平均值。与实际答案相差 10-5 以内的答案可以被接受。

整体思路是一样的，层序遍历，然后求每一层的和，然后求平均。

具体代码：

public List<Double> averageOfLevels(TreeNode root) {
        List<Double> res = new ArrayList<>();
        if(null == root) {
            return res;
        }
        Queue<TreeNode> queue = new ArrayDeque<>();
        queue.add(root);
        while(!queue.isEmpty()) {
            int size = queue.size();
            int len = size; //记录每一层的节点个数，用来求平均
            int sum = 0;
            while(size > 0) {
                TreeNode node = queue.poll();
                sum += node.val;
                if(null != node.left) {
                    queue.add(node.left);
                }
                if(null != node.right) {
                    queue.add(node.right);
                }
                size--;
            }
            Double ave = sum/len;
            res.add(ave);
        }
        return res;
    }

后面两道题每一层的循环可以改成for循环，这样更方便一点。

今天就记录到这里吧，还有很多题目要这周末做完，继续加油呀！

看到的朋友如果有疑问请留言，我一定知无不言，一起讨论！