时间戳

首页 > 解决方案 > 循环遍历未知数量的维度

c++ - 循环遍历未知数量的维度

问题描述

我想写一个循环，它可以给我一个数组的维度。我想让它可用于任何数组，它应该返回维度的大小。

int arr[3][5][10][9] ;
cout << "dimension 1: " << sizeof(arr)/sizeof(arr[0]) << endl; 
cout << "dimension 2: " << sizeof(arr[0])/sizeof(arr[0][0]) << endl; 
cout << "dimension 3: " << sizeof(arr[0][0])/sizeof(arr[0][0][0]) << endl;
cout << "dimension 4: " << sizeof(arr[0][0][0])/sizeof(arr[0][0][0][0]) << endl;
cout << "dimension 5: " << sizeof(arr[0][0][0][0])/sizeof(arr[0][0][0][0][0]) << endl;

这应该返回3,5,10,9（并且最后一条语句失败）。

所以模式看起来很清楚“每次迭代都在 .[0]之后添加arr。最后一次迭代将失败，这应该停止 while 循环。

如何“连接+评估”数组名称？

我也会感谢有关哪些测试检查“这会失败吗？”或“是否有另一个维度？”的帮助。在 C++ 中，因为我只是在学习它。

标签： c++arraysmultidimensional-array

解决方案

如果您使用的是 c++ 17 编译器，则可以使用类型特征结构std::rank，std::extent如下所示

#include <iostream>
#include <type_traits>

template<typename T>
void print_dimension(std::size_t i) {
    if (std::rank_v<T> > 0) {
        std::cout << "Dimension " << i << ":" << std::extent_v<T> << std::endl;
        print_dimension<typename std::remove_extent_t<T>>(i + 1);
    }
}

int main() {
int arr[3][5][10][9] ;
print_dimension<decltype(arr)>(1);
return 0;
}

如果您使用的是 C++ 11/14 编译器，则需要稍作修改

#include <iostream>
#include <type_traits>

template<typename T>
void print_dimension(std::size_t i) {
    if (std::rank<T>::value > 0) {
        std::cout << "Dimension " << i << ":" << std::extent<T>::value << std::endl;
        print_dimension<typename std::remove_extent<T>::type>(i + 1);
    }
}

int main() {
int arr[3][5][10][9] ;
print_dimension<decltype(arr)>(1);
return 0;
}

推荐阅读