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

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:

扫描二维码关注公众号,回复: 12237876 查看本文章

输入: 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;
    }
};

猜你喜欢

转载自blog.csdn.net/Yang_1998/article/details/108547669