LeetCode アルゴリズム バイナリ ツリー - 236. バイナリ ツリーの最近の共通祖先

目次

236. 二分木の最近の共通祖先

コード：

操作結果: 

---

バイナリ ツリーが与えられた場合、ツリー内の指定された 2 つのノードに最も近い共通の祖先を見つけます。

Baidu 百科事典における最も近い共通の祖先の定義は次のとおりです。「ルート ツリー T の 2 つのノード p と q について、最も近い共通の祖先はノード x として表されます。x は p と q の祖先であり、深さはノード x です。 x はできるだけ大きいです ( A ノードはそれ自身の祖先になることもできます)。

例 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 。なぜなら、定義上、最も近い共通の祖先ノードはノード自体である可能性があるからです。

例 3:

入力: root = [1,2]、p = 1、q = 2
出力: 1

ヒント：

• ツリー内のノードの数が [2, 105] 範囲内にあります。
• -109 <= Node.val <= 109
• すべて Node.val 互不相同 。
• p != q
• p そして q 両方とも指定されたバイナリ ツリーに存在します。

コード：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        // p=root ，则 q 在 root 的左或右子树中；
        // q=root ，则 p 在 root 的左或右子树中；
        // 即题目提示：一个节点也可以是它自己的祖先
        if(root==null||root==p||root==q) return root;
        // 不是则让左右节点继续往下递归，在本层递归看来这步是给left赋值，看看有没有p,q在左子树上
        TreeNode left=lowestCommonAncestor(root.left,p,q);
        // 与上一步一样
        TreeNode right=lowestCommonAncestor(root.right,p,q);
        // 如果left 和 right都不为空，说明此时root就是最近公共节点
        // 如果left为空，right不为空，就返回right，说明目标节点是通过right返回的，反之亦然
        if(left != null && right != null) return root;
        if(left==null) return right;
        return left;
    }
}

操作結果: