Из массива A[N][N] сформируйте массив B[N^2], "развернув" его а) по столбцам; б)по строкам

#include <iostream>
#include <iomanip>
#include <cmath>
 
// по строкам
template<typename T, std::size_t SIZE>
void copyRow(const T (&source)[SIZE][SIZE], T (&destination)[SIZE * SIZE]) {
    for (std::size_t i = 0; i < SIZE; ++i) {
        for (std::size_t j = 0; j < SIZE; ++j) {
            destination[i * SIZE + j] = source[i][j];
        }
    }
}
 
// по столбцам
template<typename T, std::size_t SIZE>
void copyColumn(const T (&source)[SIZE][SIZE], T (&destination)[SIZE * SIZE]) {
    for (std::size_t i = 0; i < SIZE; ++i) {
        for (std::size_t j = 0; j < SIZE; ++j) {
            destination[j * SIZE + i] = source[i][j];
        }
    }
}
 
// вспомогательная функция для вывода по строкам
template<typename T, std::size_t SIZE>
void print(const T (&matrix)[SIZE]) {
    std::size_t side = std::sqrt(SIZE);
    for (std::size_t i = 0; i < side; ++i) {
        for (std::size_t j = 0; j < side; ++j) {
            std::cout << std::setw(3) << matrix[i * side + j];
        }
        std::cout << '\n';
    }
}
 
int main() {
    const int size = 4;
    int source[size][size] {
            {1, 2, 3, 4},
            {5, 6, 7, 8},
            {9, 0, 1, 2},
            {3, 4, 5, 6},
    };
 
    int destination[size * size]{};
 
    // по строкам
    copyRow(source, destination);
    print(destination);
 
    std::cout << '\n';
 
    // по столбцам
    copyColumn(source, destination);
    print(destination);
 
    return 0;
}