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

题目描述

给你两棵二叉树，原始树 original 和克隆树 cloned，以及一个位于原始树 original 中的目标节点 target。

其中，克隆树 cloned 是原始树 original 的一个 副本 。

请找出在树 cloned 中，与 target 相同 的节点，并返回对该节点的引用（在 C/C++ 等有指针的语言中返回 节点指针，其他语言返回节点本身）。

注意：

你 不能 对两棵二叉树，以及 target 节点进行更改。
只能 返回对克隆树 cloned 中已有的节点的引用。
进阶：如果树中允许出现值相同的节点，你将如何解答？

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-a-corresponding-node-of-a-binary-tree-in-a-clone-of-that-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

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) {
            if(!original || !cloned || !target)  return NULL;

            //判断当前节点是否为所求
            if(original==target){
                return cloned;
            }
            TreeNode* res;
            //在左子树中找
            res=getTargetCopy(original->left,cloned->left,target);
            if(res)  return res;
            //左子树没找到，接着在右子树找
            res=getTargetCopy(original->right,cloned->right,target);
            return res;
    }
};

一次通过的感觉真不错