题目来源
题目描述
给定一棵二叉树(保证非空)以及这棵树上的两个节点对应的val值 o1 和 o2,请找到 o1 和 o2 的最近公共祖先节点。(原谅我,这截图好大一个)
解题思路
用的是递归(其实我一直对递归不太熟练来着的o(╥﹏╥)o)
公共祖先是什么样的呢?答曰:3种情况,如下图所示:
先把问题简化(以最小的树为例,扩展到整体),整个流程可分为一下几步(我们先从左子树开始找,目标结点记为o1,o2,没找到就返回-1):
-
第一种情况,我们从上到下从左子树开始找o1,找到了后再去右子树找o2,两边都找到了返回祖先结点。
解释一下这个图的大概意思,我们看以2为根结点的子树,发现他的左子树有目标o1,但是右子树没有目标o2,所以2不是祖先,往回走,看祖先的祖先,如果祖先的祖先还不是,在祖先的祖先的祖先。。。(大概是这个意思,感觉我讲的不是太好o(╥﹏╥)o,不知道我的图会不会表达地清楚一点)
-
第二种情况,左子树找到o1后,去右子树没找到o2,说明右子树不存在o2,o2只在左子树,但我们在左子树先找到的o1,不管如何o1肯定是o2的祖先节点。
-
第三种情况与第2种相反,左子树没有找到o1,两个结点都在右子树,那么o1,o2有一个是祖先节点。
代码
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param root TreeNode类
* @param o1 int整型
* @param o2 int整型
* @return int整型
*/
int lowestCommonAncestor(struct TreeNode* root, int o1, int o2 ) {
// write code here
if (root == NULL)
return -1;//未找到o1,o2
if (root->val == o1 || root->val == o2) {
return root->val;//找到o1,o2
}
int left = lowestCommonAncestor(root->left, o1, o2);
int right = lowestCommonAncestor(root->right, o1, o2);
if (left == -1)
return right;
if (right == -1)
return left;
return root->val;//运行到此处表明o1,o2均已找到
}
总结
我还是挺害怕递归的,感觉没学好,还得练练,如若有错,欢迎各位指出,谢谢!!
2024.6.14