leetcode每日一题 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

提示

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

进阶

如果树中允许出现值相同的节点，将如何解答？

解题思路

遍历 cloned 的所有结点，并比较 val ，若找到与 target 的 val 值相同的结点，则返回；否则返回 nullptr 。

本题采用二叉树的深度优先遍历DFS。

代码部分

/**
 * 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) {
        int num=target->val;
        TreeNode* res=dfs(cloned,num);
        return res;
    }
    TreeNode* dfs(TreeNode* t, int num){
        if(t==nullptr)return nullptr;   //没找到，返回nullptr
        if(t->val==num){
            return t;
        }
        TreeNode* res = dfs(t->left,num);
        if(res!=nullptr)return res;     //若左子树找到，则返回
        return dfs(t->right,num);       //找右子树
    }
};