题解 | #在二叉树中找到两个节点的最近公共祖先#

在二叉树中找到两个节点的最近公共祖先

https://www.nowcoder.com/practice/e0cc33a83afe4530bcec46eba3325116

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 * };
 */

class Solution {
public:
    /**
     * 
     * @param root TreeNode类 
     * @param o1 int整型 
     * @param o2 int整型 
     * @return int整型
     */
    int lowestCommonAncestor(TreeNode* root, int o1, int o2) {
        if (!root) {
            return -1;
        }
        if (root->val == o1 || root->val == o2) {
            return root->val;
        }
        int l = lowestCommonAncestor(root->left, o1, o2);
        int r = lowestCommonAncestor(root->right, o1, o2);
        if (l == -1) {
            return r;
        }
        if (r == -1) {
            return l;
        }
        return root->val;
    }
};

思路:树形结构一般都是递归。

* 因为节点值都是非负的,所以返回-1表示在这棵子树上没找到对应的值。

* 如果当前节点的值就是两者之一,则直接返回当前节点的值。

** 如果另一个值在当前节点的子节点,那么当前节点的值就是正确答案。

** 否则在后续逻辑进行处理。

* 递归对左右子节点求值。

** 左右子节点中有一个为-1,因为题目保证一定能找到,所以直接返回另一个子节点的结果。(另一个子节点是-1也没关系,说明不在当前递归分支)

** 左右子节点均不为-1,说明要找的两个数分别在左右子树中,那么它们的最近公共祖先就是当前节点。

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务