6154. 感染二叉树需要的总时间 - 力扣(LeetCode)
给你一棵二叉树的根节点 root
,二叉树中节点的值 互不相同 。另给你一个整数 start
。在第 0
分钟,感染 将会从值为 start
的节点开始爆发。
每分钟,如果节点满足以下全部条件,就会被感染:
- 节点此前还没有感染。
- 节点与一个已感染节点相邻。
返回感染整棵树需要的分钟数_。_
示例 1:
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-YaZaNeDp-1661077386368)(https://assets.leetcode.com/uploads/2022/06/25/image-20220625231744-1.png)]
输入: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:
[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-QnSvhCfA-1661077386369)(https://assets.leetcode.com/uploads/2022/06/25/image-20220625231812-2.png)]
输入:root = [1], start = 1
输出:0
解释:第 0 分钟,树中唯一一个节点处于感染状态,返回 0 。
提示:
- 树中节点的数目在范围
[1, 105]
内 1 <= Node.val <= 105
- 每个节点的值 互不相同
- 树中必定存在值为
start
的节点
Solution
- 最大的感染时间
- 要么是start节点的最大的子树高度
- 递归的时候,在root->val== start,记录当前root的最大子树高度
- 要么是从start出发到它的某个祖宗节点的路径长度+该祖宗节点的最大子树高度
- 假设cur为start的祖宗节点,curDepth为cur节点的深度
- 如果start在cur的左子树,那么结果为
- ans = startDepth-curDepth+r
- 如果start在cur的右子树,那么结果为
- ans = startDepth-curDepth+l;
- 如果start在cur的左子树,那么结果为
- 假设cur为start的祖宗节点,curDepth为cur节点的深度
- 要么是start节点的最大的子树高度
- 递归地取不同的节点,更新最大的ans,即可得到最终答案
- 再分析一下为什么上面的公式是可行的
- 可以证明在递归地时候cur取到了start的非祖宗节点时,ans不会超过cur取start的祖宗节点得到的ans值
- 当cur取start的某祖宗节点的另外一只子树的某节点时,上面得到的值永远小于cur取该祖宗节点得到的ans值
- 因为,此时cur节点的深度恒大于等于该祖宗节点的深度,而cur节点的最大子树高度,恒小于等于该祖宗节点的最大子树高度
- cur取start的子孙节点,上面的永远小于cur取start节点得到的值
- 因为,此时cur节点的深度永远大于等于start节点的深度,而cur节点的最大子树高度,永远小于start节点的最大子树高度
- 当cur取start的某祖宗节点的另外一只子树的某节点时,上面得到的值永远小于cur取该祖宗节点得到的ans值
- 可以证明在递归地时候cur取到了start的非祖宗节点时,ans不会超过cur取start的祖宗节点得到的ans值
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
int ans=0;
int startDepth=-1;
int dfs(TreeNode* root, int curDepth, int start){
if(!root) return 0;
if(root->val==start){
startDepth = curDepth;
}
int l = dfs(root->left, curDepth+1, start);
bool inLeft = startDepth!=-1;
int r = dfs(root->right, curDepth+1, start);
if(root->val==start){
ans = max(ans, max(l, r));
}
else if(inLeft){
ans = max(ans, startDepth - curDepth + r); // 注意加curdepth
}
else{
ans = max(ans, startDepth - curDepth + l);
}
return max(l,r)+1;
}
int amountOfTime(TreeNode* root, int start) {
dfs(root, 0, start);
return ans;
}
};