二叉树的最近公共祖先

二叉树的最近公共祖先

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        return traversal(root,p,q);
    }
    TreeNode* traversal(TreeNode*root,TreeNode*p,TreeNode*q)
    {
        if(root==q||root==p||root==NULL) return root;//如果根节点是q或者是p或者是NULL的话直接返回根节点
        TreeNode*left=traversal(root->left,p,q);//遍历左节点
        TreeNode*right=traversal(root->right,p,q);//遍历右节点
        
        if(left!=NULL&&right!=NULL) return root;//左右都不空肯定是根节点
        else if(left!=NULL&&right==NULL) return left;//如果右为空就返回左子树
        else if(left==NULL&&right!=NULL) return right;//如果左子树为空就返回右子树
        else  return NULL;//两者都为空的话就返回空
    }
};