1. 程式人生 > >Leetcode 236.二叉樹的最近公共祖先

Leetcode 236.二叉樹的最近公共祖先

二叉樹的最近公共祖先

給定一個二叉樹, 找到該樹中兩個指定節點的最近公共祖先。

百度百科中最近公共祖先的定義為:"對於有根樹 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 }