题目:
给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大(一个节点也可以是它自己的祖先)。”
思路:
1. 递归解法:(树这种结构真的太适合递归解法了):如果两个节点分别在左右子树,那么公共祖先为根结点,如果两个节点同在一边子树,那么公共祖先必在该边子树。
递归解法看起来简单,但是仔细理解起来还是不容易,设计时候也容易出错。可以把递归看成是自己调用另一个和自己功能一样,但是函数名不同的函数来理解
2.非递归解法:
题解:(C++)
1.递归解法:
struct TreeNode* lowestCommonAncestor(struct TreeNode* root, struct TreeNode* p, struct TreeNode* q) {
if(root == nullptr || root == p || root == q) return root; //递归终止条件
TreeNode* left = lowestCommonAncestor(root->left,p,q);
TreeNode* right = lowestCommonAncestor(root->right,p,q);
//看返回哪一个
return left == nullptr?right:(right == nullptr?left:root); //左边也有右边也有那就为root,不可能left和right返回都为空
}
非递归解法:
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if (root == NULL)
return NULL;
stack<TreeNode*> record;
stack<TreeNode*> help;
record.push(root);
while (!record.empty())
{
TreeNode* cur = record.top();
help.push(cur);
record.pop();
if (cur->left != NULL)
record.push(cur->left);
if (cur->right != NULL)
record.push(cur->right);
}
while (!help.empty())
{
TreeNode* node = help.top();
if (node->left != NULL && node->left->val == p->val)
p->val = node->val;
if (node->right != NULL && node->right->val == p->val)
p->val = node->val;
if (node->left != NULL && node->left->val == q->val)
q->val = node->val;
if (node->right!= NULL && node->right->val == q->val)
q->val = node->val;
if (p->val == q->val)
return p;
help.pop();
}
return NULL;
}
};
知识点:
nullptr 为空指针