最近公共祖先_leetcode No.236 二叉树的最近公共祖先
承接二叉搜索樹的最近公共祖先。
題目鏈接:
二叉樹的最近公共祖先 - 力扣(LeetCode)?leetcode-cn.com題目描述:
給定一個二叉樹, 找到該樹中兩個指定節點的最近公共祖先。
百度百科中最近公共祖先的定義為:“對于有根樹 T 的兩個結點 p、q,最近公共祖先表示為一個結點 x,滿足 x 是 p、q 的祖先且 x 的深度盡可能大(一個節點也可以是它自己的祖先)。”
例如,給定如下二叉樹: root = [3,5,1,6,2,0,8,null,null,7,4]
示例 1:
輸入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1 輸出: 3 解釋: 節點 5 和節點 1 的最近公共祖先是節點 3。示例 2:
輸入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4 輸出: 5 解釋: 節點 5 和節點 4 的最近公共祖先是節點 5。因為根據定義最近公共祖先節點可以為節點本身。說明:
- 所有節點的值都是唯一的。
- p、q 為不同節點且均存在于給定的二叉樹中。
解題思路:
在二叉搜索樹的最近公共祖先一題中,樹是一個二叉搜索樹,我們可以很容易知道應該往哪個方向去找節點的位置。但是本題就是一棵普通的數,應該怎么來求解呢?
我們肯定還是要先找到兩個節點的位置,再去判斷他倆的祖先,用 DFS 就可以了,那具體應該怎么做呢?
我們先看 p, q 是不是根節點,如果是根節點就是最近公共祖先,我們就返回根節點。如果不是,我們分別在左、右子樹中查找是否包含 p 或 q,有兩種情況:左子樹包含 p,右子樹包含q;左子樹包含 q,右子樹包含 p),那么此時的根節點還是最近公共祖先。
而如果左子樹包含 p 和 q(往右找的返回值為空),那么到 root->left 中查找,最近公共祖先在左子樹里面。
如果右子樹包含 p 和 q(往左找的返回值為空),那么到 root->right 中查找,最近公共祖先在右子樹里面。
代碼如下:
class Solution { public:TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {if (!root || root==p || root==q) {return root;}TreeNode* left = lowestCommonAncestor(root->left, p, q);TreeNode* right = lowestCommonAncestor(root->right, p, q);if(!left) return right;if(!right) return left;return root;} };如果有任何疑問,歡迎提出。如果有更好的解法,也歡迎告知。
總結
以上是生活随笔為你收集整理的最近公共祖先_leetcode No.236 二叉树的最近公共祖先的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 两千块钱带来的 希望
- 下一篇: python中zip的使用_浅谈Pyth