主要思考如下幾點:
- 如何自底向上周遊?
- 周遊整棵樹,還是周遊局部樹?
- 如何把結果傳到根結點?
關于如果遞歸函數有傳回值,如何區分是要搜尋一條邊,還是搜尋整個樹:
case1: 搜一條邊的寫法:
//若遞歸函數傳回值不為空,則立即傳回
if (遞歸函數(root->left)) return ;
if (遞歸函數(root->right)) return ;
case2: 搜尋整個樹的寫法:
left = 遞歸函數(root->left);
right = 遞歸函數(root->right);
left與right的邏輯處理;
- 自底向上尋找,即回溯,二叉樹回溯的過程即是自底向上,而後序周遊就是天然的回溯過程。
怎麼判斷一個結點是結點q和結點p的公共祖先呢:如果找到一個結點,其左子樹出現結點q
,右子樹出現結點p,或者反之,則,該結點即位結點p和q的最近公共祖先
- 确定傳回值類型:
![](https://img.laitimes.com/img/__Qf2AjLwojIjJCLyojI0JCLiIXZ05WZj91YpB3I2EzX4xSZz91ZsAzNfRHLGZkRGZkRfJ3bs92YsAjMfVmepNHL9kEWaJjVXF2cW1WUmxGSiNnVtNVQClGVF5UMR9Fd4VGdsATNfd3bkFGazxycykFaKdkYzZUbapXNXlleSdVY2pESa9VZwlHdssmch1mclRXY39CXldWYtlWPzNXZj9mcw1ycz9WL49zZuBnL0kTNyQDZyIWYwkDN5YzM4ImYhRjN5kTOzMzN4MDN5QzLc52YucWbp5GZzNmLn9Gbi1yZtl2Lc9CX6MHc0RHaiojIsJye.png)
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(root == q || root == p || root == NULL)
return root;
TreeNode* left = lowestCommonAncestor(root->left, p, q);
TreeNode* right = lowestCommonAncestor(root->right, p, q);
if(left != NULL && right != NULL) return root;
if(!left && right) return right;
else if(!right && left) return left;
else return NULL;
}
};
因為二叉搜尋樹是有序的,即可以告訴我們周遊方向,故可以使用前序周遊(這裡什麼周遊都可,因為不需要進行中間結點)
這裡是遇到符合條件的結點就傳回,故符合上面第一種情況
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
//if(root == NULL) return root;
//if(root->val >= p->val && root->val <= q->val)
//return root;
if(root->val > p->val && root->val > q->val){
TreeNode* l = lowestCommonAncestor(root->left, p, q);
if(l) return l;
}
if(root->val < p->val && root->val < q->val){
TreeNode* r = lowestCommonAncestor(root->right, p, q);
if(r) return r;
}
return root; //root節點在區間(p->val <= root->val && root->val <= root->val)或者 (root->val <= cur->val && cur->val <= root->val)中
}
};