给你两棵二叉树,原始树 original 和克隆树 cloned,以及一个位于原始树 original 中的目标节点 target。
其中,克隆树 cloned 是原始树 original 的一个 副本 。
请找出在树 cloned 中,与 target 相同 的节点,并返回对该节点的引用(在 C/C++ 等有指针的语言中返回 节点指针,其他语言返回节点本身)。
注意:
你 不能 对两棵二叉树,以及 target 节点进行更改。
只能 返回对克隆树 cloned 中已有的节点的引用。
进阶:如果树中允许出现值相同的节点,你将如何解答?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-a-corresponding-node-of-a-binary-tree-in-a-clone-of-that-tree
思路
- DFS搜索
class Solution {
public boolean flag = true;
public final TreeNode getTargetCopy(final TreeNode original, final TreeNode cloned
, final TreeNode target) {
if (original == null || !flag) {
return null;
}
if (original.equals(target)) {
flag = false;
return cloned;
}
TreeNode left = getTargetCopy(original.left, cloned.left, target);
TreeNode right = getTargetCopy(original.right, cloned.right, target);
if (left != null) {
return left;
} else if (right != null) {
return right;
}
return null;
}
}