从零学算法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;
  }
相关推荐
山登绝顶我为峰 3(^v^)343 分钟前
如何录制带备注的演示文稿(LaTex Beamer + Pympress)
c++·线性代数·算法·计算机·密码学·音视频·latex
Two_brushes.2 小时前
【算法】宽度优先遍历BFS
算法·leetcode·哈希算法·宽度优先
森焱森4 小时前
水下航行器外形分类详解
c语言·单片机·算法·架构·无人机
QuantumStack6 小时前
【C++ 真题】P1104 生日
开发语言·c++·算法
写个博客6 小时前
暑假算法日记第一天
算法
绿皮的猪猪侠6 小时前
算法笔记上机训练实战指南刷题
笔记·算法·pta·上机·浙大
hie988947 小时前
MATLAB锂离子电池伪二维(P2D)模型实现
人工智能·算法·matlab
杰克尼7 小时前
BM5 合并k个已排序的链表
数据结构·算法·链表
.30-06Springfield8 小时前
决策树(Decision tree)算法详解(ID3、C4.5、CART)
人工智能·python·算法·决策树·机器学习
我不是哆啦A梦8 小时前
破解风电运维“百模大战”困局,机械版ChatGPT诞生?
运维·人工智能·python·算法·chatgpt