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