题目描述
给定一个根为 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 之间。
- 每个结点的值都是独一无二的。
问题分析
先判断如果根节点的左子树高度等于右子树高度,那么返回根节点root;如果左子树高度大于右子树高度,说明目标节点一定在左子树中,那么用根节点的左孩子调用递归函数subtreeWithAllDeepest,然后将递归函数的返回值返回;如果左子树高度小于右子树高度,说明目标节点一定在右子树中,那么用根节点的右孩子调用递归函数subtreeWithAllDeepest,然后将递归函数的返回值返回。
代码实现
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* subtreeWithAllDeepest(TreeNode* root) {
int left = height(root->left);
int right = height(root->right);
if(left == right)
return root;
return left > right ? subtreeWithAllDeepest(root->left) : subtreeWithAllDeepest(root->right);
}
int height(TreeNode* root){
if(!root)
return 0;
return max(height(root->left), height(root->right)) + 1;
}
};