题目:
给你两棵二叉树,原始树 original
和克隆树 cloned
,以及一个位于原始树 original
中的目标节点 target
。
其中,克隆树 cloned
是原始树 original
的一个 副本 。
请找出在树 cloned
中,与 target
相同 的节点,并返回对该节点的引用(在 C/C++ 等有指针的语言中返回 节点指针,其他语言返回节点本身)。
注意:你 不能 对两棵二叉树,以及 target
节点进行更改。只能 返回对克隆树 cloned
中已有的节点的引用。
提示:
- 树中节点的数量范围为
[1, 104]
。 - 同一棵树中,没有值相同的节点。
target
节点是树original
中的一个节点,并且不会是null
。
进阶:如果树中允许出现值相同的节点,将如何解答?
思考:
普通:
“同一棵树中,没有值相同的节点” ----> 遍历克隆树中所有节点,与target节点的值相比较,若相同则直接返回节点,否则继续判断该节点的左右子节点
树的遍历 ----> 递归调用
代码如下:
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def getTargetCopy(self, original, cloned, target):
"""
:type original: TreeNode
:type cloned: TreeNode
:type target: TreeNode
:rtype: TreeNode
"""
if original is None:
return None
if cloned.val == target.val: # 直接比较克隆树的节点和目标节点的值
return cloned
left = self.getTargetCopy(original.left, cloned.left, target)
if left:
return left
return self.getTargetCopy(original.right, cloned.right, target)
提交通过:
进阶:
“树中允许出现值相同的节点” ----> 遍历原始树的所有节点,找到target节点,返回此时对应的克隆树的节点。
代码如下:
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def getTargetCopy(self, original, cloned, target):
"""
:type original: TreeNode
:type cloned: TreeNode
:type target: TreeNode
:rtype: TreeNode
"""
if original is None:
return None
if original == target: # 在原始树中找到目标节点,返回此时对应的克隆树的节点
return cloned
left = self.getTargetCopy(original.left, cloned.left, target)
if left:
return left
return self.getTargetCopy(original.right, cloned.right, target)
提交通过: