Leetcode.1379 找出克隆二叉树中的相同节点

题目链接

Leetcode.1379 找出克隆二叉树中的相同节点 easy

题目描述

给你两棵二叉树，原始树 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

提示：

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

解法：递归

递归遍历 cloned，找到与 target值相同的结点返回即可。

时间复杂度： $O(n)$

C++代码：

class Solution {
public:
    TreeNode* getTargetCopy(TreeNode* original, TreeNode* cloned, TreeNode* target) {
        if(cloned == nullptr) return nullptr;
        if(cloned->val == target->val) return cloned;
        
        auto a = getTargetCopy(original,cloned->left,target);
        if(a != nullptr) return a;
        else return getTargetCopy(original,cloned->right,target);
    }
};

Python代码：

class Solution:
    def getTargetCopy(self, original: TreeNode, cloned: TreeNode, target: TreeNode) -> TreeNode:
        if cloned == None:
            return None

        if cloned.val == target.val:
            return cloned
        a = self.getTargetCopy(original,cloned.left,target)
        if a != None:
            return a
        else:
            return self.getTargetCopy(original,cloned.right,target)