LeetCode235題:二叉搜尋樹的最近公共祖先
阿新 • • 發佈:2018-12-26
這道題要充分利用二叉搜尋樹的性質,即比某一節點大的數全在其右側,比其小的數全在其左側。
思路:
知道了二叉搜尋樹的性質後,根據其性質可知,如果p和q在節點temp的兩側的話,也就是當(p.val - temp.val)*(q.val - temp.val)<=0時(包含“=”號,因為“一個節點也可以是它自己的祖先”),則當前節點temp就是它們的最近公共祖先。否則說明p和q都在temp的某一側,如果在左側的話,那麼就讓temp = temp.left,迴圈遞進比較即可。
解法一:遞迴法
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) { if((root.val - p.val) * (root.val - q.val) <=0){ return root; }else if(root.val - p.val > 0){ root = lowestCommonAncestor(root.left,p,q); }else{ root = lowestCommonAncestor(root.right,p,q); } return root; }
解法二:迭代法
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) { while(root != null){ if(p.val < root.val && q.val < root.val) root = root.left; else if(p.val > root.val && q.val > root.val) root = root.right; else return root; } return root; }