微软高频面试题: 最近公共祖先

考虑用递归函数的思路来做:

TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q)

这个函数从root为根节点的二叉树,寻找p和q的最近公共祖先,如果最近公共祖先存在返回root

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if(root==NULL) return NULL;
        if(root==p||root==q) return root;
        TreeNode* left = lowestCommonAncestor(root->left,p,q);
        TreeNode* right = lowestCommonAncestor(root->right,p,q);
        if(left!=NULL&&right!=NULL) return root;
        else if(left==NULL) return right;
        else return left;
    }
};

当我们用递归去做这个题时不要被题目误导,应该要明确一点
这个函数的功能有三个:给定两个节点 pp 和 qq

如果 pp 和 qq 都存在,则返回它们的公共祖先;
如果只存在一个,则返回存在的一个;
如果 pp 和 qq 都不存在,则返回NULL
本题说给定的两个节点都存在,那自然还是能用上面的函数来解决

猜你喜欢

转载自blog.csdn.net/wwxy1995/article/details/108514210