Я був шокований тим, що ніхто не запропонував певної арифметично-магічної петлі для виконання роботи. Оскільки К. Ван шукає рішення без вкладених циклів , я запропоную:
double B[10][8][5];
int index = 0;
while (index < (10 * 8 * 5))
{
const int x = index % 10,
y = (index / 10) % 10,
z = index / 100;
do_something_on_B(B[x][y][z]);
++index;
}
Ну, такий підхід не є елегантним та гнучким, тому ми могли б упакувати весь процес у функцію шаблону:
template <typename F, typename T, int X, int Y, int Z>
void iterate_all(T (&xyz)[X][Y][Z], F func)
{
const int limit = X * Y * Z;
int index = 0;
while (index < limit)
{
const int x = index % X,
y = (index / X) % Y,
z = index / (X * Y);
func(xyz[x][y][z]);
++index;
}
}
Ця функція шаблону може бути виражена і у вигляді вкладених циклів:
template <typename F, typename T, int X, int Y, int Z>
void iterate_all(T (&xyz)[X][Y][Z], F func)
{
for (auto &yz : xyz)
{
for (auto &z : yz)
{
for (auto &v : z)
{
func(v);
}
}
}
}
І його можна використовувати, надаючи 3D-масив довільного розміру плюс ім'я функції, дозволяючи виведенню параметрів робити важку роботу підрахунку розміру кожного виміру:
int main()
{
int A[10][8][5] = {{{0, 1}, {2, 3}}, {{4, 5}, {6, 7}}};
int B[7][99][8] = {{{0, 1}, {2, 3}}, {{4, 5}, {6, 7}}};
iterate_all(A, do_something_on_A);
iterate_all(B, do_something_on_B);
return 0;
}
Назустріч більш загальному
Але знову ж таки, йому не вистачає гнучкості, тому що він працює лише для 3D-масивів, але, використовуючи SFINAE, ми можемо виконувати роботу для масивів довільного виміру, спочатку нам потрібна функція шаблону, яка ітераціює масиви рангу 1:
template<typename F, typename A>
typename std::enable_if< std::rank<A>::value == 1 >::type
iterate_all(A &xyz, F func)
{
for (auto &v : xyz)
{
func(v);
}
}
І ще один, який повторює масиви будь-якого рангу, виконуючи рекурсію:
template<typename F, typename A>
typename std::enable_if< std::rank<A>::value != 1 >::type
iterate_all(A &xyz, F func)
{
for (auto &v : xyz)
{
iterate_all(v, func);
}
}
Це дозволяє нам ітератувати всі елементи у всіх вимірах масиву довільних розмірів довільних розмірів.
Працювати з std::vector
Для декількох вкладених векторів рішення повторно збирає масив масиву довільного розміру довільного розміру, але без SFINAE: Спочатку нам знадобиться функція шаблону, яка повторює std::vector
s і викликає потрібну функцію:
template <typename F, typename T, template<typename, typename> class V>
void iterate_all(V<T, std::allocator<T>> &xyz, F func)
{
for (auto &v : xyz)
{
func(v);
}
}
І ще одна шаблонна функція, яка ітералізує будь-який тип векторів і називає себе:
template <typename F, typename T, template<typename, typename> class V>
void iterate_all(V<V<T, std::allocator<T>>, std::allocator<V<T, std::allocator<T>>>> &xyz, F func)
{
for (auto &v : xyz)
{
iterate_all(v, func);
}
}
Незалежно від рівня вкладеності, iterate_all
викликте версію векторів, якщо версія векторних значень не відповідає кращому, таким чином закінчуючи рекурсивність.
int main()
{
using V0 = std::vector< std::vector< std::vector<int> > >;
using V1 = std::vector< std::vector< std::vector< std::vector< std::vector<int> > > > >;
V0 A0 = {{{0, 1}, {2, 3}}, {{4, 5}, {6, 7}}};
V1 A1 = {{{{{9, 8}, {7, 6}}, {{5, 4}, {3, 2}}}}};
iterate_all(A0, do_something_on_A);
iterate_all(A1, do_something_on_A);
return 0;
}
Я думаю, що тіло функції досить просте та прямолінійне ... Цікаво, чи зможе компілятор розкрутити цю петлю (я майже впевнений, що більшість компіляторів може розгорнути перший приклад).
Дивіться демо-версію тут .
Сподіваюся, це допомагає.