假设有这么一种情况:程序中定义了一个静态数组,然后将这个数组作为参数传递给某个函数,如何在该函数中确定数组长度呢?
int arr[] = { 1, 2, 3, 4, 5 };
f(arr); // 如何在函数 f() 中确定 arr 的长度?
C 语言中遇到这种情况,一般是直接将数组长度作为参数传递给 f()
:
void f(int *arr, int n) {}
int arr[] = { 1, 2, 3, 4, 5 };
f(arr, 5);
如果 arr 是动态数组,在 C++ 中确实也只能这么做。但对于静态数组就完全没必要了,因为 arr 中本身就保留了其长度信息。
C++ 中可以利用模板函数获取静态数组参数长度,具体写法为:
template <typename T, size_t N>
void f(T (&vals)[N])
{
cout << N << endl;
}
int arr[] = { 1, 2, 3, 4, 5 };
f(arr); // T, N 分别被解析为 int, 5
还可以这么写:
template <typename T>
void f(T& arr)
{
cout << std::end(arr) - std::begin(arr) << endl;
}
int arr[] = { 1, 2, 3, 4, 5 };
f(arr); // T 被解析为 int[5]
这里 std::end(arr) - std::begin(arr)
可以写成 std::size(arr)
(需要 C++17 标准),但不应该写成 sizeof(arr) / sizeof(arr[0])
,因为当输入数组长度为0时 (异常情况,C++ 不应当出现长度为0的静态数组),其它写法会在编译期间报错,而 sizeof
的写法则不会。
在 STL 头文件 range_access.h
可以看到 std::size()
的实现:
/**
* @brief Return the size of an array.
*/
template <typename _Tp, size_t _Nm>
constexpr size_t
size(const _Tp (&)[_Nm]) noexcept
{ return _Nm; }