Очень простое заполнение матрицы змейкой по диагоналям

double **FillArr (int n)
{
    int i, j, k = 0;//Счётчики
    //Выделяем память под указатели на стркои матрицы
    double ** arr = new double *[n];
    for(i = 0; i < n; i++)
    {
        //Выделяем память под элементы i-й стркои матрицы
        arr[i] = new double [n];
    }
    //Собственно само заполнение змейкой по диагонали
    int diag = 0;
    for(diag = 0; diag < 2*n; diag++)
    {
        for(i = 0; i < n; i++)
        for(j = 0; j < n; j++)
        {
            if(diag + 1 == (i + 1) + (j + 1))
            {
                if(diag % 2)
                    arr[i][j] = k + 1;
                else
                    arr[j][i] = k + 1;
                k = k + 1;
            }
        }
    }
    return arr;
}

#include <iomanip>
#include <iostream>
using namespace std;
 
const int N = 32; //Данная константа будет определять максимальный размер матрицы
 
int main()
{
    int i, j, k = 0;//Счётчики
    int diag    = 0;//Тоже счётчик но для диагоналей
    int n = 0;//Будет содержать размер матрицы
    //Выделяем память под элементы массива
    double arr[N][N] = {0};
    cout<<"Enter size of matrix : ";cin>>n;
    if(N < n)
        cout<<"size of matrix is too large\n";
    else
    {
        //Собственно заполнение матрицы
        for(diag = 0; diag < 2*n; diag++)
        {
            for(i = 0; i < n; i++)
            for(j = 0; j < n; j++)
            {
                if(diag + 1 == (i + 1) + (j + 1))
                {
                    if(diag % 2)
                        arr[i][j] = k + 1;
                    else
                        arr[j][i] = k + 1;
                    k = k + 1;
                }
            }
        }
        //Печать заполненной матрицы
        for(i = 0; i < n; i++)
        {
            for(j = 0; j < n; j++)
                cout<<setw(2)<<arr[i][j]<<" ";
            cout<<endl;
        }
    }
    return 0;
}