C ++ Правильно освободить многомерный массив с основной непрерывной памятью - PullRequest
Новая
       40

C ++ Правильно освободить многомерный массив с основной непрерывной памятью

0 голосов
/ 31 марта 2019

У меня есть функция, которая выделяет память для 3d-массива. Цель состоит в том, чтобы память была гарантированно смежной, но также имела удобство синтаксиса a[k][j][i]: 

double *** allocate_3d(const int& sz, const int& sy, const int& sx){
    double * aMem = new double[sz*sy*sx];
    double *** array = new double**[sz];
    for(int k=0; k<sz; k++){
        array[k] = new double*[sy];
        for(int j=0; j<sy; j++){
            array[k][j]= aMem + k*sy*sx + j*sx;
        }
    }
    return array;
}

Предположим, что я вызываю такую ​​функцию 

x   = allocate_3d(wdz,wdy,wdx);

Как мне правильно освободить х? Я попробовал следующее: 

void delete_3d(double *** array, const int& sz, const int& sy, const int& sx){
    for(int k=0; k<sz; k++){    
        for(int j=0; j<sy; j++){
            delete[] array[k][j];
        }
        delete[] array[k];
    }
    delete[] array;
}

Но запуск Valgrind выдает ошибку и, похоже, указывает на то, что память не освобождена должным образом.

1 Ответ

2 голосов
/ 31 марта 2019

На самом деле должно быть так: 

void delete_3d(double *** array, const int& sz, const int& sy, const int& sx) {
    // first, restore the pointer to the actual aMem
    double * aMem = array[0][0];
    // only one dimension was allocated in the loop,
    // so only one loop should be deallocating things
    for(int k = 0; k < sz; k++) {
        delete [] array[k];
    }
    delete[] array;
    delete[] aMem;
}

Лучшее решение

Обернуть std::vector, чтобы получить и непрерывную память, идостаточно простой синтаксис доступа и в качестве бонуса нет дополнительных массивов, которые лучше для памяти, производительности и обслуживания: 

class Array3d {
public:
    Array3d(size_t sx, size_t sy, size_t sz)
        : m_sx(sx), m_sy(sy) {
        m_array.resize(sx * sy * sz);
    }

    // Can't overload operator[] with more that one parameter
    // so we'll have to do with operator()
    double &operator()(size_t x, size_t y, size_t z) {
        size_t index = x * m_sx * m_sy + y * m_sy + z;
        return m_array[index];
    }

private:
    size_t m_sx, m_sy;
    std::vector<double> m_array;
};

Пример использования: 

Array3d array(3, 3, 3);
array(0, 1, 2) = 3.14;
...