如何将指针传递给函数并在函数 C++ 中动态分配内存

我试图声明一个指针并将该指针传递给分配内存的函数。这是一个最小的例子：

#include <string>
#include <iostream>

using namespace std;

void alloc_mem(int &size, double *x);

int main()
{

        double *X;
        int imax;

        alloc_mem(imax, X);

        cout << "imax = " << imax << endl;
        for (int i = 0; i < imax; i++) {
                cout << "X = " << X[i] << endl;
        }

        delete[]X;
        return 0;

}

void alloc_mem(int &size, double *x)
{

        size = 10;
        x = new double[size];
        for (int i = 0; i < size; i++) {
                x[i] = (double)i;
        }

}

这段代码可以编译，但是当我尝试打印 X 的值时出现分段错误。我知道我没有正确地将变量传递到函数中，但我不确定如何做到这一点。我相信我正在操作 x 的副本。

另外，编写此代码是为了重现我在更大的代码中遇到的问题。

范围double *x是函数alloc_mem的局部变量。当函数结束执行时，变量将被销毁。 main 中的原始变量 X 不知道该参数做了什么，因为它是按值传递的，该值是函数中使用的它的副本。

通过指针或引用传递指针。例如

void alloc_mem(int &size, double **x);

void alloc_mem(int &size, double * &x);

void alloc_mem(int &size, double **x) 
{
   size = 10;

   *x = new double [size];

   for ( int i = 0; i < size; i++ ) ( *x )[i] = i;
}

void alloc_mem(int &size, double * &x) 
{
   size = 10;

   x = new double [size];

   for ( int i = 0; i < size; i++ ) x[i] = i;
}

对于我来说，我会按以下方式定义该函数

double * alloc_mem( int &size ) 
{
   size = 10;

   x = new double [size];

   for ( int i = 0; i < size; i++ ) x[i] = i;

   return x;
}

如果在调用函数之前知道大小，那么它可以写得更简单

double * alloc_mem( int size ) 
{
   x = new double [size];

   for ( int i = 0; i < size; i++ ) x[i] = i;

   return x;
}

考虑到该循环

   for ( int i = 0; i < size; i++ ) x[i] = i;

可以替代标准算法std::iota例如

std::iota( x, x + size, 0.0 );