在做算法题时,有一题需要知道输入数组的行数和列数,学习后将此方法记录下:
#include<vector>
int row = array.size();//行数
int col = array[0].size();//列数(假设每行的列数都相同)
解释:
在 C++ 中,`std::vector` 是一个模板容器类,它动态地管理数组的大小,提供了对元素的安全访问,并且能够报告它当前所含有的元素数量。对于二维 vector(即 `vector<vector<int>>` 类型),它实际上是一个由 vector 组成的 vector,我们可以通过以下方式获取行数和列数:
获取行数
使用 `array.size()` 方法可以获取外层 vector 的大小,也就是行数。在 `std::vector` 类中,`size()` 方法返回容器中当前存储的元素数量,这里的元素是 `vector<int>`,每个元素代表一行数据。
获取列数
若要获取列数(即每行的元素数量),可以使用 `array[i].size()`,其中 `i` 是指定行的索引。这个方法返回内层 vector(即第 `i` 行)的大小,也就是这一行中的列数。理论上,二维 vector 中的每一行都可以有不同的列数(即每个内层 vector 的长度可以不同),因此正确的做法是对每一行分别调用 `.size()` 方法以获取准确的列数。
例如如下所示的二维 vector:
std::vector<std::vector<int>> array = {
{1, 2, 3},
{4, 5},
{6, 7, 8, 9}
};
- 使用 `array.size()` 将返回 `3`,因为有三行。
- 使用 `array[0].size()` 将返回 `3`,因为第一行有三个元素。
- 使用 `array[1].size()` 将返回 `2`,因为第二行有两个元素。
- 使用 `array[2].size()` 将返回 `4`,因为第三行有四个元素。
这样就可以非常灵活地处理各种大小和形状的二维数据结构咯!