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 之间。
• 每个结点的值都是独一无二的。

问题分析

先判断如果根节点的左子树高度等于右子树高度，那么返回根节点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;
    }
};