使用一个unique_ptr来管理一个三维数组

Question

我使用以下方式分配了一个三维空间：

board = new char**[depth];
    for (int d = 0; d < depth; d++)
    {
        board[d] = new char*[rows];
        for (int i = 0; i < rows; i++) {
            board[d][i] = new char[cols];
            for (int j = 0; j < cols; j++) {
                board[d][i][j] = 'k';
            }
        }
    }

现在我想用一个unique_ptr来管理它，但我不知道如何使用unique_ptr公开的接口来启动这样一个结构。

Answer 1

表面上，你可以这样写：

std::unique_ptr<std::unique_ptr<std::unique_ptr<char[]>[]>[]>board = 
    std::make_unique<std::unique_ptr<std::unique_ptr<char[]>[]>[]>(depth);
for (int d = 0; d < depth; d++)
{
    board[d] = std::make_unique<std::unique_ptr<char[]>[]>(rows);
    for (int i = 0; i < rows; i++) {
        board[d][i] = std::make_unique<char[]>(cols);
        for (int j = 0; j < cols; j++) {
            board[d][i][j] = 'k';
        }
    }
}

这或多或少会按您预期的方式工作。

但是神圣的意大利面条代码，蝙蝠侠！那是一些非常草率的代码！

你可以像这样使用 std::vector 来简化它，这是一种更自然的结构，用于表示数组的数组：

std::vector<std::vector<std::vector<char>>> board;
board.resize(depth);
for (int d = 0; d < depth; d++)
{
    board[d].resize(rows);
    for (int i = 0; i < rows; i++) {
        board[d][i].resize(cols);
        for (int j = 0; j < cols; j++) {
            board[d][i][j] = 'k';
        }
    }
}

但是考虑到你写的本质上是一个 3D 矩形矩阵，这样写它仍然没有多大意义。相反，让我们这样做：

size_t get_index(size_t i, size_t j, size_t k, size_t rows, size_t cols, size_t depth) {
    if(i > rows || j > cols || k > depth) throw std::runtime_error("Out of bounds!");
    return k * rows * cols + j * rows + i;
}

size_t get_size(size_t rows, size_t cols, size_t depth) {
    return rows * cols * depth;
}

int main() {
    size_t rows = 15, cols = 10; depth = 5;
    std::vector<char> board(get_size(rows, cols, depth));
    for(size_t d = 0; d < depth; d++) {
        for(size_t r = 0; r < rows; r++) {
            for(size_t c = 0; c < cols; c++) {
                board[get_index(r, c, d, rows, cols, depth)] = 'k';
            }
        }
    }

}

您可以考虑将所有这些打包到一个 Matrix class 中，它自己处理所有这些翻译。然后你就可以编写如下代码：

int main() {
    size_t rows = 15, cols = 10; depth = 5;

    Matrix<char> board(rows, cols, depth);
    for(size_t d = 0; d < board.get_depth(); d++) {
        for(size_t r = 0; r < board.get_rows(); r++) {
            for(size_t c = 0; c < board.get_cols(); c++) {
                board(r, c, d) = 'k';
            }
        }
    }

}

它更干净，更容易维护，如果你使你的 Matrix class 通用，可重复用于其他事情。

使用一个unique_ptr来管理一个三维数组

Using a unique_ptr to manage a three dimensional array

c++

arrays

memory-management

unique-ptr