Расположить максимальные элементы строк вдоль побочной диагонали матрицы - C (СИ)

#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#define N 5
 
int main(void)
{
    int A[N][N], i, j, iMax, tmp;
 
    srand(time(NULL));
    for (i = 0; i < N; i++)
    {
        for (j = 0; j < N; j++)
        {
            A[i][j] = rand() % 10;
            printf("%d ", A[i][j]);
        }
        printf("\n");
    }
 
    printf("\nAfter:\n");
    for (i = 0; i < N; i++)
    {
        iMax = 0;
        for (j = 1; j < N; j++)
            if (A[i][iMax] < A[i][j])
                iMax = j;
 
        if (iMax != N - 1 - i)
        {
            tmp = A[i][N - 1 - i];
            A[i][N - 1 - i] = A[i][iMax];
            A[i][iMax] = tmp;
        }
 
        for (j = 0; j < N; j++)
            printf("%d ", A[i][j]);
        printf("\n");
    }
 
    return 0;
}