Блог

Я хочу передать массив определенной функции на c . Я написал следующий код:

 #define nmpart 50

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

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

        distances(dis);
}
  

Этот код работает нормально, когда nmpart<800 примерно. Проблема в том, что я хотел бы
чтобы иметь больший массив, допустим, для nmpart = 50000. Я прочитал на этом форуме, что
для решения этой проблемы можно использовать динамическое распределение. Мне непонятно, как можно
В этой ситуации я использую динамическое распределение. Не могли бы вы дать мне несколько советов.

Многие люди (включая меня) никогда бы не написали этот код профессионально (или как-то иначе), но в основном он отвечает на ваш вопрос. Смотрите ниже, почему я бы никогда не написал это профессионально.

 #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 amp;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
}
  

Вместо этого вы можете использовать std::vector. Это позволяет выполнять динамическое распределение.

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

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

    distances(dis);
}