最近公共祖先结点——中等难度

Posted 2021-06-07 我是一颗大萝北

一、题目描述

二、问题分析

看到二叉树的题一般就用递归求解。
还是以根节点的视角来看，
（1）如果根节点的左分支有o1或o2，右分支有o2或o1，那么最近共同祖先结点就是根节点，return 根节点的值
（2）如果根节点的左分支有o1和o2，右分支什么都没有，那么最近共同祖先结点根节点在根节点的左分支，return 左分支return回来的值
（3）如果根节点的右分支有o1和o2，左分支什么都没有，那么最近共同祖先结点根节点在根节点的右分支，return 右分支return回来的值
（4）如果左分支和右分支都没有，return 0

那么怎样让递归函数返回共同祖先的值呢？
首先，这个递归函数得是int类型的；其次，当o1和o2只在左分支或右分支时，回溯返回的得是已经找到的最近公共祖先结点的值，这样才能在最后return回来最近公共祖先结点的值

三、初版代码

/**
 * 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) {
        // write code here
          return ifhave(root,o1,o2);
        
    }
    int ifhave(TreeNode* root,int o1, int o2){
        if(root==nullptr)
              return 0;
        if(root->val==o1)
              return o1;
        else if(root->val==o2)
              return o2;
        int left=ifhave(root->left,o1,o2);
        int right=ifhave(root->right,o1,o2);
        if(left>=1&&right==0)  //root左边有，右边没有，那么祖先已经找到了，就是左边返回的值
              return left;
        if(left==0&&right>=1)  //同理，root左边没有，右边有，那么祖先已经找到了，就是右边返回的值
              return right;
        if(left&&right>=1)  //root的左边也有，右边也有，那么就是root
              return root->val;
        return 0; //左右都没有的话，return 0
    }
};