Заполнение матрицы змейкой по диагонали из правого верхнего угла

/*
  Дано натуральное число n (1 <= n <= 10)
  Заполнить квадратную матрицу послеовательностью чисел 0, 1, 2 ... n^2 - 1
  по заданной схеме
  Схема: змейка по диагонали(слева).
*/
 
#include "pch.h"
#include <iostream>
#include <iomanip>
#include <locale.h>
#include <math.h>
using namespace std;
 
//Функция выделяет динамическую память под квадратную матрицу размера nxn и возвращает ее адрес
int ** CreateMatrix(int n)
{
    int ** a = new int *[n];
    for (int i = 0; i < n; i++)
        a[i] = new int[n];
    return a;
}
 
// Функция выводит квадратную матрицу a размера n*n на консоль
void PrintMatrix(int ** a, int n)
{
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < n; j++)
            cout << setw(4) << a[i][j];
        cout << endl;
    }
}
 
// Функция вычисляет для элемента змейки k его координаты в
// матрице размера n x n, то есть a[i,j] = k
void Coordinates(int k, int n, int & i, int & j)
{
    int s, min, k1, i1, j1;
    if ((k <= n * (n + 1) / 2 - 1)) //  Верхний треугольник
    {
        s = floor((-1 + sqrt(1 + 8 * k)) / 2); // номер диагонали
        min = s * (s + 1) / 2; // минимальный элемент диагонали s
        if (s % 2 == 0) //  четная диагональ - снизу вверх
            i = s - (k - min);
        else //  нечетная диагональ
            i = k - min;
        j = s - i;
    }
    else
    {
        k1 = n * n - 1 - k; //  симметричный элементу k элемент из верхнего треугольника
        s = floor((-1 + sqrt(1 + 8 * k1)) / 2); // номер диагонали
        min = s * (s + 1) / 2;  // минимальный элемент диагонали s
        if (s % 2 == 0) // четная диагональ - снизу вверх
            i1 = s - (k1 - min);
        else // нечетная диагональ
            i1 = k1 - min;
        j1 = s - i1;
 
        i = n - 1 - i1;
        j = n - 1 - j1;
    }
    return;
}
 
 
//Функция заполняет матрицу а по схеме - диагональная змейка
// Способ заполнения - заполнение матрицы значениями с вычислением
// их координат по математическим соотношениям
void FillMatrix(int **& a, int n)
{
    int i, j;
    for (int k = 0; k <= n * n - 1; k++)
    {
        Coordinates(k, n, i, j);
        a[i][j] = k;
    }
    return;
}
 
int main()
{
    setlocale(LC_ALL, "russian");
    int n;
    do {
        cout << "Введите размер квадратной матрицы" << endl;
        cin >> n;
    } while ((n < 1) || (n > 10));
    int ** a = CreateMatrix(n);
    FillMatrix(a, n);
    cout << "Полученная матрица: " << endl;
    PrintMatrix(a, n);
 
    system("pause");
}