给定一个根为 root 的二叉树,每个结点的深度是它到根的最短距离。
如果一个结点在整个树的任意结点之间具有最大的深度,则该结点是最深的。
一个结点的子树是该结点加上它的所有后代的集合。
返回能满足“以该结点为根的子树中包含所有最深的结点”这一条件的具有最大深度的结点。
示例:
输入:[3,5,1,6,2,0,8,null,null,7,4]
输出:[2,7,4]
解释:
我们返回值为 2 的结点,在图中用黄色标记。
在图中用蓝色标记的是树的最深的结点。
输入 "[3, 5, 1, 6, 2, 0, 8, null, null, 7, 4]" 是对给定的树的序列化表述。
输出 "[2, 7, 4]" 是对根结点的值为 2 的子树的序列化表述。
输入和输出都具有 TreeNode 类型。
提示:
树中结点的数量介于 1 和 500 之间。
每个结点的值都是独一无二的。
t
思路:两次dfs,第一次求出树的最大深度已经在最大深度的节点个数;第二次dfs找到尽可能深的树根使得其子树包含了所有最深节点。
class Solution {
int num;
int depth;
TreeNode ans;
public TreeNode subtreeWithAllDeepest(TreeNode root) {
num=0;
depth=0;
ans=null;
dfs1(root,0);
dfs2(root,depth);
return ans;
}
private void dfs1(TreeNode root,int d) {
if(root==null) {
if(depth==d) num++;
else if(depth<d) num=1;
depth=Math.max(depth, d);
return;
}
dfs1(root.left,d+1);
dfs1(root.right,d+1);
}
private int dfs2(TreeNode root,int d) {
if(root==null) {
if(d==0) return 1;
return 0;
}
int res=dfs2(root.left,d-1)+dfs2(root.right,d-1);
if(res==num && ans==null) ans=root;
return res;
}
}