从零学算法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;
  }
相关推荐
Wendy14418 小时前
【线性回归(最小二乘法MSE)】——机器学习
算法·机器学习·线性回归
拾光拾趣录8 小时前
括号生成算法
前端·算法
渣呵8 小时前
求不重叠区间总和最大值
算法
拾光拾趣录9 小时前
链表合并:双指针与递归
前端·javascript·算法
好易学·数据结构9 小时前
可视化图解算法56:岛屿数量
数据结构·算法·leetcode·力扣·回溯·牛客网
香蕉可乐荷包蛋10 小时前
AI算法之图像识别与分类
人工智能·学习·算法
chuxinweihui10 小时前
stack,queue,priority_queue的模拟实现及常用接口
算法
tomato0910 小时前
河南萌新联赛2025第(一)场:河南工业大学(补题)
c++·算法
墨染点香10 小时前
LeetCode Hot100【5. 最长回文子串】
算法·leetcode·职场和发展
甄卷12 小时前
李沐动手学深度学习Pytorch-v2笔记【08线性回归+基础优化算法】2
pytorch·深度学习·算法