題目:
給定一個二叉樹, 找到該樹中兩個指定節點的最近公共祖先。百度百科中最近公共祖先的定義爲:“對於有根樹 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 爲空指針