从零学算法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;
  }
相关推荐
秋说14 分钟前
【PTA数据结构 | C语言版】两枚硬币
c语言·数据结构·算法
qq_5139704425 分钟前
力扣 hot100 Day37
算法·leetcode
不見星空1 小时前
leetcode 每日一题 1865. 找出和为指定值的下标对
算法·leetcode
我爱Jack1 小时前
时间与空间复杂度详解:算法效率的度量衡
java·开发语言·算法
DoraBigHead3 小时前
小哆啦解题记——映射的背叛
算法
Heartoxx3 小时前
c语言-指针与一维数组
c语言·开发语言·算法
孤狼warrior3 小时前
灰色预测模型
人工智能·python·算法·数学建模
京东云开发者3 小时前
京东零售基于国产芯片的AI引擎技术
算法
chao_7894 小时前
回溯题解——子集【LeetCode】二进制枚举法
开发语言·数据结构·python·算法·leetcode
十盒半价4 小时前
从递归到动态规划:手把手教你玩转算法三剑客
javascript·算法·trae