将数组传递给C++中的函数

Question

我想将一个数组传递给 C++ 中的某个函数。我写了以下代码：

#define nmpart 50

void distances( double (&dis)[3][nmpart][nmpart] )
{
    ... compute distances which are allocated in "dis"
}

double energy()
{
        double dis[3][nmpart][nmpart];

        distances(dis);
}

当 nmpart

Answer 1

现在，很多人（包括我自己）永远不会专业地（或以其他方式）编写此代码，但它基本上回答了您的问题。请参阅下文，了解为什么我从不专业地写这篇文章。

#define nmpart 50  

void distances( const double ***dist) 
{     
    ... compute distances which are allocated in "dis" 
}  

double energy()
{ 
     double ***dis;

     // allocate dis
     dis = new double**[3];
     for(int i=0;i<3;++i)
     {
         dis[i] = new double*[nmpart];
         for(int j=0;j<nmpart;++j)
         {
              dis[i][j] = new double[nmpart];
         }
     }

     // code where you populate dis
     // HERE

     distances(dis);

     // deallocate dis
     for(int i=0;i<3;++i)
     {
         for(int j=0;j<nmpart;++j)
         {
              delete [] dis[i][j];
         }
         delete [] dis[i];
     }
     delete [] dis;

} 

基本上，我永远不会写这个（除了用于教学目的），因为如果在这个函数的某个地方抛出异常，那么它就会泄漏内存。最好的办法是将堆分配和释放（分别使用new 和delete）包装在一个类中，并将该类放在energy() 函数内的本地堆栈中。

理想情况下你会这样做：

#define nmpart 50    

void distances( const distClass &dis)  
{          
   ... compute distances which are allocated in "dis"  
}

double energy() 
{  
    distClass dis;
    // this allocate wraps the for loops with the new []'s above
    dis.Allocate(3, nmpart , nmpart);  

    // populate dis HERE

    distances(dis);

    // when dis goes out of scope its destructor
    // is called, which wraps the for loops with the delete []'s
}

Answer 2

您可以使用 std::vector 代替。这允许动态分配。

void distances( std::vector< std::vector< std::vector<double> > > const& dis )
{
    ... compute distances which are allocated in "dis"
}

double energy()
{
    std::vector< std::vector< std::vector<double> > > dis;

    distances(dis);
}