leetcode_1379 找出克隆二叉树中的相同节点

134 阅读1分钟

要求

给你两棵二叉树,原始树 original 和克隆树 cloned,以及一个位于原始树 original 中的目标节点 target。

其中,克隆树 cloned 是原始树 original 的一个 副本 。

请找出在树 cloned 中,与 target 相同 的节点,并返回对该节点的引用(在 C/C++ 等有指针的语言中返回 节点指针,其他语言返回节点本身)。

  • 注意:

    • 你 不能 对两棵二叉树,以及 target 节点进行更改。
    • 只能 返回对克隆树 cloned 中已有的节点的引用。 进阶:如果树中允许出现值相同的节点,你将如何解答?

示例 1:

image.png

输入: tree = [7,4,3,null,null,6,19], target = 3
输出: 3
解释: 上图画出了树 original 和 cloned。target 节点在树 original 中,用绿色标记。答案是树 cloned 中的黄颜色的节点(其他示例类似)。

示例 2:

image.png

输入: tree = [7], target =  7
输出: 7

示例 3:

image.png

输入: tree = [8,null,6,null,5,null,4,null,3,null,2,null,1], target = 4
输出: 4

示例 4:

image.png

输入: tree = [1,2,3,4,5,6,7,8,9,10], target = 5
输出: 5

示例 5:

image.png

输入: tree = [1,2,null,3], target = 2
输出: 2

提示:

  • 树中节点的数量范围为 [1, 10^4] 。
  • 同一棵树中,没有值相同的节点。
  • target 节点是树 original 中的一个节点,并且不会是 null 。

核心代码

class Solution:
    def getTargetCopy(self, original: TreeNode, cloned: TreeNode, target: TreeNode) -> TreeNode:
        if not original:
            return None
        if target is original:
            return cloned
        return self.getTargetCopy(original.left,cloned.left,target) or self.getTargetCopy(original.right,cloned.right,target)

image.png

解题思路:题干描述众多,其实就是我们在两棵树中进行搜索,当搜索到相对应的值的时候,直接将克隆树的节点直接返回即可,深度优先遍历即可解决。