手摸手提桶跑路——LeetCode1302. 层数最深叶子节点的和

60 阅读1分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第27天,点击查看活动详情

题目描述

给你一棵二叉树的根节点 root ,请你返回 层数最深的叶子节点的和 。

示例 1:

微信截图_20220817123240.png

输入: root = [1,2,3,4,5,null,6,7,null,null,null,null,8]
输出: 15

示例 2:

输入: root = [6,7,8,2,7,1,3,9,null,1,4,null,null,null,5]
输出: 19

提示:

  • 树中节点数目在范围 [1, 104] 之间。
  • 1 <= Node.val <= 100

解题思路——队列

题目意思很明确了,就是求最后一层的和,最后一层也叫叶子节点,就是没有子节点的节点。

这题我们可以用队列的方式进行广度优先遍历,当遍历到最后一次循环时,就是最后一层节点了。最后只需要单独的将最后一层的节点值进行累加输出即可。

那么怎么使用队列进行广度优先遍历呢?

拿树 [1,2,3,4,5,6,7] 来说。

微信截图_20220817131613.png

那么我们一想,4、5、6、7 肯定是最深的节点了吧,我们来验证一下。

  1. 首先我们将根节点 root 放入队列 queue 中,我们去循环这个队列,直到 quote.length === 0 为止。
  2. 每次遍历,我们都将 queue 出队一个元素,这里就是根节点出队了,是不是也相当于我们已经遍历了根节点了?那么既然我们是广度优先遍历,接来下要遍历的是不是左右节点了?所以接下来我们将 root.leftroot.right 分别入队,进入下一次循环。

这边有个注意点,这会我们的队列只有一个根节点 root,所以只要把 root.leftroot.right 入队就好了,那么对于 root.leftroot.right 来说,这层一共有 2 + 2 个子节点,为了能实现广度优先遍历,我们是不是要将 root.leftroot.left.leftroot.left.right 入队之后,继续入队 root.left.leftroot.left.right,所以这里需要一个循环来执行出队和入队子节点操作。

while(queue.length) {
    for(let i=0; i<queue.length; ++i) {
        let node = queue.shift();
        node.left && queue.push(node.left);
        node.right && queue.push(node.right);
    }
}
  1. 循环到 queue.length === 0 时,就完成了广度优先遍历。

题解

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number}
 */
var deepestLeavesSum = function(root) {
    const queue = [root], res = [];

    while(queue.length) {
        let lens = queue.length;
        let total = [];
        for(let i=0; i<lens; ++i) {
            let node = queue.shift();
            total.push(node.val);
            node.left && queue.push(node.left);
            node.right && queue.push(node.right);
        }
        res.push(total)
    }
    return res.slice(-1)[0].reduce((f, c) => f + c, 0);
};

微信截图_20220817123533.png

解题思路——DFS

深度优先遍历(DFS)是优先竖向搜索,而广度优先遍历(BFS)是优先横向搜索。

这道题用 DFS 的思路就是在递归的情况下去找到最大的层级,然后根据层级来判断是否是最后一层,从而进行和的统计。

题解

var deepestLeavesSum = function(root) {
    let maxLevel = 0;
    let sum = 0;
    const dfs = (node, level) => {
        if (node == null) {
            return;
        }
        if (level > maxLevel) {
            maxLevel = level;
            sum = node.val;
        } else if (level === maxLevel) {
            sum += node.val;
        }
        dfs(node.left, level + 1);
        dfs(node.right, level + 1);
    }
    dfs(root, 0);
    return sum;
};