给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。
百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
说明:
所有节点的值都是唯一的。
p、q 为不同节点且均存在于给定的二叉树中。
个人思路在于先求出root到两个节点的路径上所需要路过的节点,分别用vecLeft和vecRight来存储,然后再找出两个数组中第一个相同值,代码如下:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
static const auto init = []() {
std::ios::sync_with_stdio(false);
std::cin.tie(nullptr);
return nullptr;
}();
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if (p == root || q == root) // p或q当中有一个是根节点,则根节点为两者的公共祖先
return root;
// // 先查找从根节点到p所经过的所有节点
vector<TreeNode*> vecLeft;
this->findTreeNodePath(root, p, vecLeft);
vector<TreeNode*> vecRight;
this->findTreeNodePath(root, q, vecRight);
int size1 = vecLeft.size();
int size2 = vecRight.size();
if (size1 == size2)
{
for (int i = 0; i < size1; i++)
{
if (vecLeft[i] == vecRight[i])
{
return vecLeft[i];
}
}
}
else
{
for (int i = 0; i < size1; i++)
{
for (int j = 0; j < size2; j++)
{
if (vecRight[j] == vecLeft[i])
return vecLeft[i];
}
}
}
return root;
}
// 从根节点到指定节点之间的路径,用数组存储
TreeNode* findTreeNodePath(TreeNode* root, TreeNode* targetNode, vector<TreeNode*> & vec)
{
if (root == NULL || targetNode == NULL)
return NULL;
if (root != targetNode)
{
TreeNode* node = this->findTreeNodePath(root->left, targetNode, vec);
if (node != NULL)
{
vec.push_back(root);
}
else
{
node = this->findTreeNodePath(root->right, targetNode, vec);
if (node != NULL)
vec.push_back(root);
}
return node;
}
else
{
vec.push_back(root);
return root;
}
return NULL;
}
};