JAVA程序设计:具有所有最深结点的最小子树(LeetCode:865)

给定一个根为 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;
    } 
}
原创文章 1111 获赞 194 访问量 25万+

猜你喜欢

转载自blog.csdn.net/haut_ykc/article/details/105682388