1. 题目
给你两棵二叉树,原始树 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 解题
- 循环方式的二叉树遍历,两棵树同步进行即可
class Solution {
public:TreeNode* getTargetCopy(TreeNode* original, TreeNode* cloned, TreeNode* target) {stack<TreeNode*> stk1, stk2;TreeNode *tp1, *tp2;while(original || !stk1.empty()){while(original){stk1.push(original);original = original->left;}while(cloned){stk2.push(cloned);cloned = cloned->left;}tp1 = stk1.top();tp2 = stk2.top();stk1.pop();stk2.pop();if(tp1 == target)return tp2;original = tp1->right;cloned = tp2->right;}return NULL;}
};
872 ms 164.6 MB