二叉树系列之一(求最近公共父节点)

首先给出数据结构：

* struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };

常见的函数声明为：

//给定两个节点判断
TreeNode* lowAncestor(TreeNode* root,TreeNode* p,TreeNode* q);
TreeNode* lowAncestor(TreeNode* root,int v1,v2);
//给定数组或者链表判断
TreeNode* lowAncestor(TreeNode* root,vector<int> sV);
TreeNode* lowAncestor(TreeNode* root,vector<TreeNode*> sV);

对于求父节点也是很常见的问题，该如何思考这个问题呢？

1.考虑到最近，我们是不是可以想到，假如节点本身就是另一个节点的父节点，这时是不是就是最近的父节点了？所以我们判断的条件成为：

if(root->val==v1||root->val==v2)
{
    return root;
}

2.判断条件给出来了，那我们开始思考方法：递归方法，那么上述的判断条件完全可以作为递归返回的过程。
无规律的暴力遍历

 TreeNode* lowestCommonAncestor(TreeNode* root, int v1,int v2) {
    
    return find(root,v1,v2);
 }
TreeNode* find(TreeNode* root,int v1,v2)
{
    if(root==nullptr)//递归函数的返回条件，也即函数调用栈结束压栈的条件
    {
        return nullptr;
    }
    //符合上述判断条件最近的情况
    if(root->val==v1&&root->val==v2)
    {
        return root;
    }
    //如果v1的节点和v2的节点不是直接的父子关系，那么我们就需要去左右子树中去遍历
    //因为，树未必符合BST的格式，所以要暴力搜索
    TreeNode*left=find(root->left,v1,v2);
    TreeNode*right=find(root->right,v1,v2);
    //root的左右子树分别找到了，那说明root为最近的ancestor
    if(left!=nullptr&&right!=nullptr)
    {
        return root;
    }
    //如果上述两个都没找到，所有v1，v2全在左子树或者全在右子树，所以要递归了
    return left==nullptr?right:left;
}

基于二叉树的遍历剪枝
 

 TreeNode* lowestCommonAncestor(TreeNode* root, int v1,int v2) {
    
    return find(root,v1,v2);
 }
TreeNode* find(TreeNode* root,int v1,v2)
{
    if(root==nullptr)//递归函数的返回条件，也即函数调用栈结束压栈的条件
    {
        return nullptr;
    }
    if(root->val>v2&&root->val>v1)//v1，v2节点在右子树上
    {
        return find(root->left,v1,v2);
    }
    else if(root->val<v2&&root->val<v1)//v1，v2节点在左子树上
    {
        return find(root->left,v1,v2);
    }
    //v1，v2节点在root节点的左右子树上
    return root;
}

对于放多个节点或者节点值的函数

 TreeNode* lowestCommonAncestor(TreeNode* root, vector<int>& nodes) {
    unordered_set<int>st;
    for(auto& node:nodes)
    {
        st.insert(node->val);
    }
    return find(root,st);
 }

TreeNode* find(TreeNode* root,int v1,v2)
{
    if(root==nullptr)//递归函数的返回条件，也即函数调用栈结束压栈的条件
    {
        return nullptr;
    }
    //判断输入的节点就是最近的根节点+前序遍历，这么遍历得到的结果一定是根节点
    if(st.find(root->val)!=st.end())
    {
        return root;
    }
    //如果v1的节点和v2的节点不是直接的父子关系，那么我们就需要去左右子树中去遍历
    //因为，树未必符合BST的格式，所以要暴力搜索
    TreeNode*left=find(root->left,v1,v2);
    TreeNode*right=find(root->right,v1,v2);
    //root的左右子树分别找到了，那说明root为最近的ancestor
    if(left!=nullptr&&right!=nullptr)
    {
        return root;
    }
    //如果上述两个都没找到，所有v1，v2全在左子树或者全在右子树，所以要递归了
    return left==nullptr?right:left;
}