《程序员面试金典》(第六版)习题:仅为记录一下以加强印象,不为商业用途,如有侵权请联系删除。以下源码和解释参考了书中源码以及解释。
这里我们假设Node结构为二叉树的一个节点,算法的工作就是完全复制一个与原二叉树一样的二叉树。这里完全就是一颗二叉树的中序遍历,访问根节点然后构造和根节点一样的根节点并将原根节点和新建立的根节点映射起来,然后再对原数据结构的左右子树进行递归调用。一个简单的例子如图1所示。
class Node
{
private:
int data;
public:
Node(int d = 0)
{
data = d;
left=nullptr;
right=nullptr;
}
int getData()
{
return data;
}
Node* left;
Node* right;
};
Node* copyRecursive(Node *cur, map<Node*, Node*> & nodeMap)
{
if (cur == nullptr)
return nullptr;
std::map<Node*, Node*>::iterator i= nodeMap.find(cur);
//当Node结构为二叉树且允许指向祖先节点时下面的两句代码才会被调用。
if (i != nodeMap.end())
i->second;
Node* node = new Node(cur->getData());
nodeMap.insert(pair<Node*, Node*>(cur,node));
node->left = copyRecursive(cur->left, nodeMap);
node->right = copyRecursive(cur->right, nodeMap);
return node;
}
Node* copyStructure(Node* root)
{
map<Node*, Node*> nodeMap;
return copyRecursive(root,nodeMap);
}
![](https://i-blog.csdnimg.cn/blog_migrate/da5b9d1e8e17e105c3714e6ee3341a28.png)