leetcode-1379-找出克隆二叉树中的相同节点-medium_java 找出克隆二叉树中的相同节点

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

示例 2:

输入: tree = [7], target = 7 输出: 7

示例 3:

输入: tree = [8,null,6,null,5,null,4,null,3,null,2,null,1], target = 4
输出: 4

示例 4:

输入: tree = [1,2,3,4,5,6,7,8,9,10], target = 5 输出: 5

示例 5:

输入: tree = [1,2,null,3], target = 2 输出: 2

提示：

树中节点的数量范围为 [1, 10^4] 。 同一棵树中，没有值相同的节点。 target 节点是树 original
中的一个节点，并且不会是 null 。

解题思路

同时层序遍历（bfs）original和cloned二叉树，当在original中找到target的时候，返回对应此时cloned树指向的那个结点即可

C++代码
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */

class Solution {
public:
    TreeNode* getTargetCopy(TreeNode* original, TreeNode* cloned, TreeNode* target) {
        queue<TreeNode*>q1;
        queue<TreeNode*>q2;

        q1.push(original);
        q2.push(cloned);
        while(!q1.empty()){
            TreeNode * tmp1 = q1.front();
            TreeNode * tmp2 = q2.front();
            q1.pop();
            q2.pop();

            if(tmp1 == target)
                return tmp2;
            
            if(tmp1->left){
                q1.push(tmp1->left);
                q2.push(tmp2->left);
            }
            if(tmp2->right){
                q1.push(tmp1->right);
                q2.push(tmp2->right);
            }
        }
        return NULL;
    }
};
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40