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

最新推荐文章于 2024-10-13 18:33:13 发布

边界流浪者

最新推荐文章于 2024-10-13 18:33:13 发布

阅读量182

点赞数

分类专栏：水题AC记 LeetCode C++ 文章标签： leetcode c++

本文链接：https://blog.csdn.net/qq_16542775/article/details/119300328

版权

LeetCode 同时被 3 个专栏收录

251 篇文章 0 订阅

订阅专栏

C++

241 篇文章 1 订阅

订阅专栏

水题AC记

206 篇文章 0 订阅

订阅专栏

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

示例 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 。

我这算是水了一题把。

/**
 * 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) {
        dfs(original, cloned, target);
        return cloneTarget;
    }


    void dfs(TreeNode* original, TreeNode* cloned, TreeNode* target) {
        if (original == NULL || cloned == NULL) {
            return;
        }
        if (original == target) {
            cloneTarget = cloned;
            return;
        }

        if (original->left && cloned->left) {
            dfs(original->left, cloned->left, target);
        }

        if (original->right && cloned->right) {
            dfs(original->right, cloned->right, target);
        }

    }

private:
    TreeNode* cloneTarget;

};