leetcode - 最深叶节点的最近公共祖先

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

回想一下:

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

输入: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 ,最近公共祖先是它自己。
 

提示:

给你的树中将有 1 到 1000 个节点。
树中每个节点的值都在 1 到 1000 之间。
每个节点的值都是独一无二的。

题解:

        最深叶节点的最近公共祖先,首先要找到最深也节点,且一定存在。对于节点cur, 如果cur的左节点和cur的右节点的深度相同,那么cur一定为最近公共祖先。如果cur的左节点深度大于右节点深度,那么最近公共祖先一定在左节点,否则相反。

        本题使用dfs进行递归遍历,为了避免重复的计算每个节点的深度,使用几个map进行保存每个节点的深度来进行剪枝。

/**
 * 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 mp[1001];
    TreeNode* lcaDeepestLeaves(TreeNode* root) {
        int l = getdeep(root->left);
        int r = getdeep(root->right);
        if(l == r) return root;
        else if(l > r) return lcaDeepestLeaves(root->left);
        else return lcaDeepestLeaves(root->right);
    }

    int getdeep(TreeNode* root) {
        if(!root) {
            return 0;
        }
        if(mp[root->val] != 0) return mp[root->val];

        return mp[root->val] = max(getdeep(root->left), getdeep(root->right)) + 1;
        
    }
};

猜你喜欢

转载自blog.csdn.net/txgANG/article/details/120039969