难度中等23收藏分享切换为英文接收动态反馈
给你两棵二叉树,原始树 original
和克隆树 cloned
,以及一个位于原始树 original
中的目标节点 target
。
其中,克隆树 cloned
是原始树 original
的一个 副本 。
请找出在树 cloned
中,与 target
相同 的节点,并返回对该节点的引用(在 C/C++ 等有指针的语言中返回 节点指针,其他语言返回节点本身)。
注意:
- 你 不能 对两棵二叉树,以及
target
节点进行更改。 - 只能 返回对克隆树
cloned
中已有的节点的引用。
进阶:如果树中允许出现值相同的节点,你将如何解答?
示例 1:
输入: tree = [7,4,3,null,null,6,19], target = 3 输出: 3 解释: 上图画出了树 original 和 cloned。target 节点在树 original 中,用绿色标记。答案是树 cloned 中的黄颜色的节点(其他示例类似)。
示例 2:
输入: tree = [7], target = 7 输出: 7
示例 3:
输入: tree = [8,null,6,null,5,null,4,null,3,null,2,null,1], target = 4 输出: 4
示例 4:
输入: tree = [1,2,3,4,5,6,7,8,9,10], target = 5 输出: 5
示例 5:
输入: tree = [1,2,null,3], target = 2 输出: 2
提示:
- 树中节点的数量范围为
[1, 10^4]
。 - 同一棵树中,没有值相同的节点。
target
节点是树original
中的一个节点,并且不会是null
。
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def getTargetCopy(self, original: TreeNode, cloned: TreeNode, target: TreeNode) -> TreeNode:
queue = collections.deque()
queue.append(cloned)
while queue:
for i in range(len(queue)):
node = queue.popleft()
if node.val == target.val:
return node
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
return None