Leetcode 1123. 最深叶节点的最近公共祖先

Leetcode 1123. 最深叶节点的最近公共祖先

题目

给你一个有根节点的二叉树，找到它最深的叶节点的最近公共祖先。

回想一下：

叶节点 是二叉树中没有子节点的节点
树的根节点的 深度 为 0，如果某一节点的深度为 d，那它的子节点的深度就是 d+1
如果我们假定 A 是一组节点 S 的 最近公共祖先，S 中的每个节点都在以 A 为根节点的子树中，且 A 的深度达到此条件下可能的最大值。
 

注意：本题与力扣 865 重复：https://leetcode-cn.com/problems/smallest-subtree-with-all-the-deepest-nodes/

示例 1：

输入：root = [3,5,1,6,2,0,8,null,null,7,4]
输出：[2,7,4]
解释：
我们返回值为 2 的节点，在图中用黄色标记。
在图中用蓝色标记的是树的最深的节点。
注意，节点 6、0 和 8 也是叶节点，但是它们的深度是 2 ，而节点 7 和 4 的深度是 3 。

示例 2：

输入：root = [1]
输出：[1]
解释：根节点是树中最深的节点，它是它本身的最近公共祖先。

示例 3：

输入：root = [0,1,3,null,2]
输出：[2]
解释：树中最深的叶节点是 2 ，最近公共祖先是它自己。

思路

• 由于这道题目求得是最深叶子节点的最近公共祖先， 我的想法是肯定首选递归
• 求出左子树的高度和右子树的高度， 如果高度相等， 当前节点必然是答案
• 如果左子树高度比右子树高度高， 答案肯定在左子树， 知道了这样的思想， 就可以写代码了

代码 —— c++

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int get_height(TreeNode* root) {
        if (!root) return 0;
        else return max(get_height(root->left), get_height(root->right)) + 1;
    }

    void dfs(TreeNode*& res, TreeNode* root) {
        if (!root) return;
        else {
            int left_height = get_height(root->left);
            int right_height = get_height(root->right);

            if (left_height == right_height) {
                res = root;
            } else if (left_height < right_height) {
                dfs(res, root->right);
            } else {
                dfs(res, root->left);
            }
        }
    }

    TreeNode* lcaDeepestLeaves(TreeNode* root) {
        TreeNode* res = NULL;
        dfs(res, root);
        return res;
    }
};