从零学算法1379

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

给你两棵二叉树，原始树 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

dfs：将两树以相同方式遍历，如果找到 original 中的 target 返回此时对应的 cloned 即可，否则继续对左右子树进行搜索

java 复制代码

  public final TreeNode getTargetCopy(final TreeNode original, final TreeNode cloned, final TreeNode target) {
      if(cloned==null)return null;
      // 找到对应节点则直接返回
      if(original==target)return cloned;
      TreeNode originLeft = original.left;
      TreeNode clonedLeft = cloned.left;
      TreeNode originRight = original.right;
      TreeNode clonedRight = cloned.right;
      // 继续以相同方式对左右子树搜索
      TreeNode left = getTargetCopy(originLeft,clonedLeft,target);
      TreeNode right = getTargetCopy(originRight,clonedRight,target);
      return left==null?right:left;
  }

bfs：或是两树以相同方式层序遍历，用两个队列遍历两棵树，当 q1 弹出节点等于 target 时，此时 q2 弹出节点即为最终答案。

java 复制代码

  public final TreeNode getTargetCopy(final TreeNode original, final TreeNode cloned, final TreeNode target) {
      Queue<TreeNode> q1 = new ArrayDeque<>();
      Queue<TreeNode> q2 = new ArrayDeque<>();
      q1.offer(original);
      q2.offer(cloned);
      while(!q1.isEmpty()){
          TreeNode node1 = q1.poll(), node2 = q2.poll();
          if(node1==target)return node2;
          if(node1.left!=null){
              q1.offer(node1.left);
              q2.offer(node2.left);
          }
          if(node1.right!=null){
              q1.offer(node1.right);
              q2.offer(node2.right);
          }
      }
      return null;
  }