235. Lowest Common Ancestor of a Binary Search Tree

[kitano-kazuki]

https://leetcode.com/problems/lowest-common-ancestor-of-a-binary-search-tree/description/

[sasanquaneuf]

平衡ではない、ずっと子が片方にしかない木もBSTの条件は満たしうるので、O(log N)は平衡な場合に限ります。

[kitano-kazuki]

たしかに、言及するべきでした

[sasanquaneuf]

面接を想定すると、このアプローチに結構時間がかかっているので、アプローチを考えたときにBSTという条件を使っているか？（ただの木で考えていないか？）というメタ的な検討はあって良いかもしれません。メタ的な検討5〜10分ぐらいでCode2-2に至れていたら、そちらの方が時間コストが良いですね。

[kitano-kazuki]

間違いないですね。code2-2はステップ2で他の解法ないかなって思ったら割とすぐ出てきたものなので、メタ的な視点持てると良かったです。
自分自身、解法思いついた時に飛びつく癖がありそうな気はしています

[sasanquaneuf]

left, rightを使わないといけないのはpとqの左右が一定しないからだと思うので、p.val > q.valのときに入れ替えると、単純な比較で書けると思います。
これを単純な比較にして、あと実装の幅として再帰にしてみると以下のようにも書けます。

class Solution:
    def lowestCommonAncestor(self, root: 'TreeNode', p: 'TreeNode', q: 'TreeNode') -> 'TreeNode':
        if p.val > q.val:
            return self.lowestCommonAncestor(root, q, p)
        if p.val <= root.val <= q.val:
            return root
        if q.val < root.val:
            return self.lowestCommonAncestor(root.left, p, q)
        if root.val < p.val:
            return self.lowestCommonAncestor(root.right, p, q)
        raise Exception('something went wrong')

[kitano-kazuki]

複雑になってしまっている部分を特定して固定できるとシンプルになりますね。