题目:
给你一棵二叉树的根节点 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
的节点
解答:
![](https://img-blog.csdnimg.cn/direct/269e8614ea20485a99283bdcf65412cb.png)
代码:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
private TreeNode startNode;
private final Map<TreeNode,TreeNode> fa=new HashMap<>();
public int amountOfTime(TreeNode root, int start) {
dfs(root,null,start);
return maxDepth(startNode,startNode);
}
private void dfs(TreeNode node,TreeNode from,int start){
if(node==null) return;
fa.put(node,from);//记录每个节点的父节点
if(node.val==start){
startNode=node;//找到start
}
dfs(node.left,node,start);
dfs(node.right,node,start);
}
private int maxDepth(TreeNode node,TreeNode from){
if(node==null){
return -1;//注意这里是-1,因为start的深度为0
}
int res=-1;
if(node.left!=from){
res=Math.max(res,maxDepth(node.left,node));
}
if(node.right!=from){
res=Math.max(res,maxDepth(node.right,node));
}
if(fa.get(node)!=from){
res=Math.max(res,maxDepth(fa.get(node),node));
}
return res+1;
}
}