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

// Направление dir=0 - вверх, =1 Вниз-направо по диагонали, 
// = 2 - параво, =3 - вверх-налево по диагонали
for(int i=4, j=0,dir=0, ii=0; ii<25; ii++) {
  A[i][j] = ii+1;
  if(dir==0) {
    i--;
    dir=1;
 }
 else if (dir==1) {
   j++;
   if (i==4) 
     dir = 2;
   else
      i++;
 }
 else if (dir==2) {
    j++;
    dir = 3;
 } 
 else {
   i--;
   if (j==0)
     dir = 0;
   else
     j--;
 }
}

#define ROWS 5
#define COLS 5
 
int main()
{
    int matr[ROWS][COLS], direction = 1;
 
    for( int i = ROWS - 1, j = 0, value = 1; value <= ROWS * COLS; i -= direction, j -= direction, value++ )
    {
        if( i < 0 || i > ROWS - 1 || j < 0 || j > COLS - 1 )
            direction = -direction;
 
        if     ( j > COLS - 1 )    i -= 2, j--;
        else if( i < 0        )    j += 2, i++;
        else if( j < 0        )    j++;
        else if( i > ROWS - 1 )    i--;
 
        matr[i][j] = value;
    }
 
    for( int i = 0; i < ROWS; i++ )
    {
        for( int j = 0; j < COLS; j++ )
            cout << setw( 3 ) << matr[i][j];
        cout << endl;
    }
    return 0;
}