面试中用C++写了两个分配和释放int数组的函数

Question

我被要求编写两个函数来在 C++ 中分配和释放 int 数组。

int* allocate(int size){
    return new int[size];
}

void deallocate(int *pt){
    delete pt;
    pt = NULL;
}

我想到了上面两个函数

有谁知道在 C++ 中将函数写入 allocate/deallocate int 数组有更好的方法吗？

Answer 1

I doubt there is a better way

这不是关于更好的方法，而是关于正确性。

使用

delete [] pt;

因为pt 是一个数组!

此外，正如 thorsan 所建议的，您将 pt 设置为 NULL，但在 deallocate() 之外是看不到的，请自行查看：

#include <iostream>

using namespace std;

int* allocate(int size) {
    return new int[size];
}

void deallocate(int *pt) {
    delete [] pt;
    pt = NULL;
}

int main() {
        int* pt = allocate(5);
        deallocate(pt);
        cout << (pt == NULL ? "NULL" : "not NULL") << endl;
        return 0;
}

输出：

gsamaras@pythagoras:~$ g++ -Wall main.cpp 
gsamaras@pythagoras:~$ ./a.out 
not NULL

为避免这种情况，只需传递一个引用，像这样：

void good_deallocate(int*& pt) {
        delete [] pt;
        pt = NULL;
}

您还可以在第一个函数中检查分配是否成功，如下所示：

int* good_allocate(int size) {
        try {
                return new int[size];
        }
        catch(std::bad_alloc&) {
                cerr << "shit\n";
                return NULL;
        }
        // OR as Dietmar Kühl suggested
        /*
        if (int* rc = new(std::nothrow) int[size]) {
                return rc; 
        } else {
                // handle error
        }
        */
}

灵感来自 How to check memory allocation failures with new operator?

Answer 2

if (int* rc = new(std::nothrow) int[size]) {
        return rc; 
} else {
        cout << "New failed, unable to allocate" << object_name << 
                ", size needed:" << sizeof( int ) * size << "." << endl;
        exit..., return... whatever, just don't use rc.
}

许多评论（此处和其他地方）是关于 "why bother to handle?" 您的程序无法运行，new/alloc 错误不会发生在当今的大型（虚拟）机器上，等等。想一想具有未检测到的 'new' 错误（或任何错误，就此而言）的程序是如何失败的。对于 'new'，稍后您通常会看到访问冲突。它可能与最初失败的 statement/variable(s).

面试中用C++写了两个分配和释放int数组的函数

Write two functions to allocate and deallocate int array in C++ in the interview

c++

pointers

new-operator

delete-operator

dynamic-allocation