首页 > 文章列表 > C++ 函数返回多维数组时如何确定形状?

C++ 函数返回多维数组时如何确定形状?

多维数组 c++
434 2024-04-23

为了确定 C++ 函数返回的多维数组的形状,使用以下步骤:使用 size() 确定数组的行数。使用 shape()[0] 或 arr[0].size() 确定数组的列数。

C++ 函数返回多维数组时如何确定形状?

使用 Size-Shape 特性确定 C++ 函数返回多维数组的形状

当从 C++ 函数返回多维数组时,需要确定数组的形状,以便正确处理数组元素。以下是如何使用 size() 和 shape() 方法确定形状:

#include <iostream>
#include <vector>

using namespace std;

vector<vector<int>> create_2d_array(int rows, int cols) {
  vector<vector<int>> arr(rows, vector<int>(cols));
  return arr;
}

int main() {
  // 创建一个 3x4 的二维数组
  vector<vector<int>> arr = create_2d_array(3, 4);

  // 获取数组的形状
  int rows = arr.size();
  int cols = arr[0].size();

  // 访问数组元素
  for (int i = 0; i < rows; i++) {
    for (int j = 0; j < cols; j++) {
      cout << arr[i][j] << " ";
    }
    cout << endl;
  }

  return 0;
}

输出:

0 0 0 0
0 0 0 0
0 0 0 0

在这个示例中,create_2d_array 函数返回一个 3x4 的二维数组,size()shape() 方法用于确定数组的形状,从而可以正确访问数组元素。