Leetcode 236.二叉樹的最近公共祖先
阿新 • • 發佈:2018-12-31
二叉樹的最近公共祖先
給定一個二叉樹, 找到該樹中兩個指定節點的最近公共祖先。
百度百科中最近公共祖先的定義為:"對於有根樹 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 為不同節點且均存在於給定的二叉樹中。
本題與LeetCode 235的區別就是由二叉查詢樹變為了二叉樹,即資料從有序變為了無序,那麼就不能通過root的值和兩個節點的值的大小關係來劃分查詢區域。在題235的程式碼基礎做了些許變動,同樣使用遞迴搜尋的方法,當root非空時,對root->left和root->right分別進行搜尋。若搜尋結果均非空,說明兩個節點分別位於左右子樹之中,LCA則為root;若只有一個結果為空,則LCA是另一個非空的節點;若結果均空,則返回NULL。
1 /** 2 * Definition for a binary tree node. 3 * public class TreeNode { 4 * int val; 5 * TreeNode left; 6 * TreeNode right; 7 * TreeNode(int x) { val = x; } 8 * } 9 */ 10 class Solution { 11 public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {12 if(root==null||root==p||root==q) return root; 13 TreeNode left=lowestCommonAncestor(root.left,p,q); 14 TreeNode right=lowestCommonAncestor(root.right,p,q); 15 if(left!=null&&right!=null) 16 return root; 17 if(left==null) 18 return right; 19 if(right==null) 20 return left; 21 return null; 22 } 23 }