|
| 1 | +package Week_03 |
| 2 | + |
| 3 | +//给定一个二叉树, 找到该树中两个指定节点的最近公共祖先。 |
| 4 | +// |
| 5 | +// 百度百科中最近公共祖先的定义为:“对于有根树 T 的两个结点 p、q,最近公共祖先表示为一个结点 x,满足 x 是 p、q 的祖先且 x 的深度尽可能大( |
| 6 | +//一个节点也可以是它自己的祖先)。” |
| 7 | +// |
| 8 | +// 例如,给定如下二叉树: root = [3,5,1,6,2,0,8,null,null,7,4] |
| 9 | +// |
| 10 | +// |
| 11 | +// |
| 12 | +// |
| 13 | +// |
| 14 | +// 示例 1: |
| 15 | +// |
| 16 | +// 输入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 1 |
| 17 | +//输出: 3 |
| 18 | +//解释: 节点 5 和节点 1 的最近公共祖先是节点 3。 |
| 19 | +// |
| 20 | +// |
| 21 | +// 示例 2: |
| 22 | +// |
| 23 | +// 输入: root = [3,5,1,6,2,0,8,null,null,7,4], p = 5, q = 4 |
| 24 | +//输出: 5 |
| 25 | +//解释: 节点 5 和节点 4 的最近公共祖先是节点 5。因为根据定义最近公共祖先节点可以为节点本身。 |
| 26 | +// |
| 27 | +// |
| 28 | +// |
| 29 | +// |
| 30 | +// 说明: |
| 31 | +// |
| 32 | +// |
| 33 | +// 所有节点的值都是唯一的。 |
| 34 | +// p、q 为不同节点且均存在于给定的二叉树中。 |
| 35 | +// |
| 36 | +// Related Topics 树 |
| 37 | +// 👍 870 👎 0 |
| 38 | + |
| 39 | + |
| 40 | +//leetcode submit region begin(Prohibit modification and deletion) |
| 41 | +/** |
| 42 | + * Definition for a binary tree node. |
| 43 | + */ |
| 44 | +class TreeNode(var _value: Int) { |
| 45 | + var value: Int = _value |
| 46 | + var left: TreeNode = null |
| 47 | + var right: TreeNode = null |
| 48 | +} |
| 49 | + |
| 50 | + |
| 51 | +object Leet236 { |
| 52 | + def lowestCommonAncestor(root: TreeNode, p: TreeNode, q: TreeNode): TreeNode = { |
| 53 | + trait Result |
| 54 | + case class Found(ancestor: TreeNode) extends Result |
| 55 | + case class AtLarge(missing: Set[TreeNode]) extends Result |
| 56 | + def getState(o: TreeNode, target: Set[TreeNode]): Result = { |
| 57 | + if (o == null) AtLarge(target) |
| 58 | + else { |
| 59 | + val tNew = target - o |
| 60 | + if (tNew.isEmpty) { |
| 61 | + Found(o) |
| 62 | + } |
| 63 | + else getState(o.left, tNew) match { |
| 64 | + case Found(x) if target.contains(o) => Found(o) |
| 65 | + case Found(x) => Found(x) |
| 66 | + case AtLarge(t1) => |
| 67 | + getState(o.right, t1) match { |
| 68 | + case Found(x) if target == t1 => Found(x) |
| 69 | + case Found(x) => Found(o) |
| 70 | + case AtLarge(t2) => AtLarge(t2) |
| 71 | + } |
| 72 | + } |
| 73 | + } |
| 74 | + } |
| 75 | + |
| 76 | + getState(root, Set(p, q)) match { |
| 77 | + case Found(o) => o |
| 78 | + case _ => root |
| 79 | + } |
| 80 | + } |
| 81 | +} |
| 82 | + |
| 83 | +//leetcode submit region end(Prohibit modification and deletion) |
| 84 | + |
0 commit comments