Question

我从GeeksForGeeks找到了这段代码，内容涉及从树的数组表示形式将节点导入树中。

我尝试实现自己的版本，但无法正常运行。然后我意识到必须返回Node *（在我自己的版本中：treeNode *）。

如果我已经在函数中修改了指针（根），为什么必须这样做？

来自GeeksForGeeks：

// Function to insert nodes in level order 
Node* insertLevelOrder(int arr[], Node* root, int i, int n) 
{ 
    // Base case for recursion 
    if (i < n) 
    { 
        Node* temp = newNode(arr[i]); 
        root = temp; 

        // insert left child 
        root->left = insertLevelOrder(arr, 
                   root->left, 2 * i + 1, n); 

        // insert right child 
        root->right = insertLevelOrder(arr, 
                  root->right, 2 * i + 2, n); 
    } 
    return root; 
}

我自己的版本：

treeNode* import_treeNode(treeNode* root, int nodes[], int curr_i, int size){
    if (curr_i < size){
        treeNode newNode = treeNode(nodes[curr_i]);
        root = &newNode;
        root->left = import_treeNode(root->left, nodes, 2 * curr_i + 1, size);
        root->right = import_treeNode(root->right, nodes, 2 * curr_i + 2, size);
    }
    return root;
}

我的版本不起作用，因为它没有成功返回设置了值的treeNode。我可以验证treeNode构造函数是否起作用。问题出在root-> left = import_treeNode没有成功设置左节点，右节点也是如此。

Answer 1

您将返回一个指针，该指针的生存期已经结束。

treeNode* import_treeNode(treeNode* root, int nodes[], int curr_i, int size){
    if (curr_i < size){
        // The lifetime of "newNode" begins when its constructor completes.
        treeNode newNode = treeNode(nodes[curr_i]);
        root = &newNode;
        // Now root points at newNode.
        root->left = import_treeNode(root->left, nodes, 2 * curr_i + 1, size);
        root->right = import_treeNode(root->right, nodes, 2 * curr_i + 2, size);
        // The above are the same as just assigning newNode.left and newNode.right.

        // Here the destructor is invoked for "newNode", and its lifetime ends.
    }
    // Returns a dangling pointer, which once pointed at newNode.
    // Using that pointer in almost any way results in undefined behavior.
    return root;
}

最好使用std::unique_ptr<treeNode>或std::shared_ptr<treeNode>，它们更容易意外地作为悬挂指针离开。或者，如果必须使用原始指针，则至少需要使用treeNode关键字创建一个new对象，以便其生存期可以比函数调用长。

如果在函数中修改了指针，为什么要返回指向对象的指针呢？

1 个答案: