给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个节点 p、q,最近公共祖先表示为一个节点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
示例 1:
输入:root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1
输出:3
解释:节点 5 和节点 1 的最近公共祖先是节点 3 。
解题思路:先寻找每个节点的祖先,然后遍历每个节点的祖先,找到第一个相同的点(我看到这个题的第一想法是这个样子的,还有用递归写出来很简洁。)
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int find(TreeNode* root,TreeNode* p,vector<TreeNode*> &v)//找到每个节点的祖先
{
if(root==NULL)
{
return 0;
}
if(root==p)
{
v.push_back(root);
return 1;
}
//如果一颗二叉树他的左孩子或右孩子有要查询的节点,该节点就是祖先节点
if(find(root->left,p,v)||find(root->right,p,v))
{
v.push_back(root);
return 1;
}
else
{
return 0;
}
}
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
vector<TreeNode*> v1;
vector<TreeNode*> v2;
find(root,p,v1);
find(root,q,v2);
//从后往前遍历两个数组,如果哪个时候不相等了,后一个就是两个节点的公共祖先。
int i=v1.size()-1;
int j=v2.size()-1;
while(i>=0&&j>=0)
{
if(v1[i]!=v2[j])
{
break;
}
i--;
j--;
}
return v1[i+1];
}
};