Заполнить массив по соответствующей схеме

#include <stdio.h>
#include <stdlib.h>
 
void print_matrix(int** A,int n)
{
    for (int i=0;i<n;i++)
    {
        for (int j=0;j<n;j++)
            printf("%4d ",A[i][j]);
        printf("\n");
    }
}
 
void fill_matrix(int** A,int n)
{
    int direction=-1;//определяет, вверх или вниз идём по матрице
    int cur_number=0;
    for (int i=0;i<n;i++)
    {
        int j;
        if (direction<0) j=n-1;
        else j=i;
        for (;;j+=direction)
        {
            if (j<i)
            {
                direction=1;
                break;
            }
            if (j>n-1)
            {
                direction=-1;
                break;
            }
            A[j][i]=cur_number;
            cur_number++;
        }
    }
}
 
int main()
{
    int **matrix,n=5;
    matrix=(int**)malloc(n*sizeof(int*));
    for (int i=0;i<n;i++)
        matrix[i]=(int*)calloc(n,sizeof(int));
    fill_matrix(matrix,n);//заполняем матрицу
    print_matrix(matrix,n);//выводим матрицу на экран
    for (int i=0;i<n;i++)
        free(matrix[i]);
    free(matrix);
}