1. 程式人生 > >LeetCode235題:二叉搜尋樹的最近公共祖先

LeetCode235題:二叉搜尋樹的最近公共祖先

這道題要充分利用二叉搜尋樹的性質,即比某一節點大的數全在其右側,比其小的數全在其左側。

思路:

知道了二叉搜尋樹的性質後,根據其性質可知,如果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;
    }