LeetCode解法汇总2385. 感染二叉树需要的总时间

目录链接:

力扣编程题-解法汇总_分享+记录-CSDN博客

GitHub同步刷题项目:

https://github.com/September26/java-algorithms

原题链接:. - 力扣(LeetCode)


描述:

给你一棵二叉树的根节点 root ,二叉树中节点的值 互不相同 。另给你一个整数 start 。在第 0 分钟,感染 将会从值为 start 的节点开始爆发。

每分钟,如果节点满足以下全部条件,就会被感染:

  • 节点此前还没有感染。
  • 节点与一个已感染节点相邻。

返回感染整棵树需要的分钟数

示例 1:

输入:root = [1,5,3,null,4,10,6,9,2], start = 3
输出:4
解释:节点按以下过程被感染:
- 第 0 分钟:节点 3
- 第 1 分钟:节点 1、10、6
- 第 2 分钟:节点5
- 第 3 分钟:节点 4
- 第 4 分钟:节点 9 和 2
感染整棵树需要 4 分钟,所以返回 4 。

示例 2:

输入:root = [1], start = 1
输出:0
解释:第 0 分钟,树中唯一一个节点处于感染状态,返回 0 。

提示:

  • 树中节点的数目在范围 [1, 105] 内
  • 1 <= Node.val <= 105
  • 每个节点的值 互不相同
  • 树中必定存在值为 start 的节点

解题思路:

首先,构造递归方法,方法的返回值为当前节点是否包含start节点。如果找到start节点,则计算这个节点为出发点的最远路径,这是备选值1。上层递归方法中,分左右节点查询。如果左节点中包含start节点,路径为当前节点右节点的最远路径+start节点和当前节点的层级差,这是备选值2。右节点包含也是类似的原理。持续往上递归,找出所有备选值2。求出所有备选值中最大值即可。

代码:

class Solution {
    int leafLevel = 0;
    int maxPath = 0;

    public int amountOfTime(TreeNode root, int start) {
        searchStartNode(root, start, 0);
        return maxPath;
    }


    boolean searchStartNode(TreeNode node, int start, int level) {
        if (node == null) {
            return false;
        }
        if (node.val == start) {
            maxPath = searchLevel(node, 0) - 1;
            leafLevel = level;
            return true;
        }
        if (searchStartNode(node.left, start, level + 1)) {
            maxPath = Math.max(maxPath, leafLevel - level + searchLevel(node.right, 0));
            return true;
        }
        if (searchStartNode(node.right, start, level + 1)) {
            maxPath = Math.max(maxPath, leafLevel - level + searchLevel(node.left, 0));
            return true;
        }
        return false;
    }

    int searchLevel(TreeNode node, int level) {
        if (node == null) {
            return level;
        }
        level++;
        return Math.max(searchLevel(node.left, level), searchLevel(node.right, level));
    }
}

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-04-28 17:32:02       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-04-28 17:32:02       101 阅读
  3. 在Django里面运行非项目文件

    2024-04-28 17:32:02       82 阅读
  4. Python语言-面向对象

    2024-04-28 17:32:02       91 阅读

热门阅读

  1. 机器学习之K-medians聚类

    2024-04-28 17:32:02       28 阅读
  2. [LLM]大模型训练DeepSpeed(三)

    2024-04-28 17:32:02       27 阅读
  3. 企业微信私有化部署对接oauth2.0

    2024-04-28 17:32:02       28 阅读
  4. 计算机网络复习(第一章概述)

    2024-04-28 17:32:02       27 阅读
  5. 2024.4.28每日一题

    2024-04-28 17:32:02       33 阅读