二元搜索樹賦值運算符內存泄漏

Question

所以我似乎無法找出爲什麼我泄漏內存，任何人都可以幫忙嗎？我已經實現了一個二叉搜索樹operator=:

BinTree& BinTree::operator=(const BinTree &other) 
{ 
    if (other.root == NULL) 
     root = NULL; 
    else 
    { 
     copyHelper(root, other.root); 
    } 
    return *this; 

} 

copyHelper：

void BinTree::copyHelper(Node *&current, const Node *other) 
{ 
    if (other == NULL) 
     current = NULL; 
    else 
    { 
     current = new Node; 
     current->data = other->data; 
     copyHelper(current->left, other->left); 
     copyHelper(current->right, other->right); 
    } 
} 

我明白current = new Node;是在泄漏發生的事情，但我的程序崩潰，如果我嘗試在該行之前做delete current; 。

這裏是我的析構函數：

BinTree::~BinTree() 
{ 
    destroyRecursive(root); 

} 

void BinTree::destroyRecursive(Node *node) 
{ 
    if (node) 
    { 
     destroyRecursive(node->left); 
     destroyRecursive(node->right); 
     delete node; 
    } 
} 

Answer 1

由於您使用原始指針，你必須手動刪除分配的內存。當您設置current到NULL

void BinTree::copyHelper(Node *&current, const Node *other) 
{ 
    if (other == NULL) 
    { 
     delete current; // w/o this line next 
     current = NULL; // will cause memory leak (if not managed somewhere else) 
    } 
    else 
    { 
     current = new Node; 
     current->data = other->data; 
     copyHelper(current->left, other->left); 
     copyHelper(current->right, other->right); 
    } 
} 

Answer 2

可憐的代碼結構，你會泄漏內存。 C++爲您提供了完成這項任務的所有工具，但您只需要很少或沒有任何優勢。您需要刪除當前樹的的根目錄，在運算符功能中的，，然後copy-construct新的節點;你需要修復你的析構函數。

1. 的遞歸複製助手應該在Node類拷貝構造函數：

   Node::Node(const Node &other) : left(0), right(0) 
   { 
       this->data = other.data; // This must be a deep copy 
       if (other.left) 
        this->left = new Node(other.left); 
       if (other.right) 
        this->right = new Node(other.right); 
   } 
   

2. 這是由BinTree賦值運算符稱爲：

   BinTree& BinTree::operator=(const BinTree &other) 
   { 
       delete root; 
       this->root = 0; 
       if (other.root) 
        this->root = new Node(other.root); 
       return *this; 
   } 
   

3. 你不需要遞歸銷燬方法。所有你在BinTree析構函數需要的是delete root;，併爲Node析構函數：

   Node::~Node() 
   { 
       delete left; 
       delete right; 
   } 
   

4. 的data的複製必須是深拷貝。例如，如果它是C字符串，請在Node的析構函數中使用strdup(),和free()。如果它是一個類，那個類必須有一個賦值操作符。