Двумерные массивы. Главная диагональ и произведение среднего арифметического значения элементов.

#include <iostream.h>
#define I 5     
#define J 5     
 
            
 
void input(int [I][J],char);
void output(int[I][J],char);
 
void main()
 {
 int d[I][J], s[I][J];
 input(d,'d');
 input(s,'s');
 output(d,'d');
 output(s,'s');
 }
 
void input(int x[I][J],char mas)
  {
  int i,j;
  cout << "\n Вводим массив " << mas << ": \n";
  for(i=0;i<I;i++)
   for(j=0;j<J;j++)
    {
    cout << mas << "["<<i<<"]["<<j<<"]=";
    cin >> x[i][j];
    }
  }
 
void output (int x[I][J],char mas)
 {
 int i,j;
 cout << "\n Исходный массив " << mas << ": \n";
 for(i=0;i<I;i++)
   {
   cout << "\n";
   for(j=0;j<J;j++)
     {
     cout.width(7);
     cout << x[i][j];
     }
     cout << endl;
   }
 }

for (i = 0; i < n; ++i)
{   // Заполняем строку с номером i
    for (j = 0; j < i; ++j)
    {
    A[i][j] = 2;    // Сначала пишем значение 2 элементам ниже диагонали
    }
    A[i][j] = 1;        // После завершения предыдущего цикла i==j, пишем 1
    for (++j; j < n; ++j)  // Цикл начинаем с увеличения j на 1
    {
        A[i][j] = 0;    // Записываем значение 0 элементам выше диагонали
    }
}

float average_lower(int A[I][J])
{
    int sum = 0;
    int count = 0;
    for (int i=1; i<I; i++)
        for (int j=0; j<i; j++)
        {
            sum = sum + A[i][j];
            count++;
        }
    return sum/static_cast<float>(count);
}
 
float average_higher(int A[I][J])
{
    int sum = 0;
    int count = 0;
    for (int i=0; i<I-1; i++)
        for (int j=i+1; j<I; j++)
        {
            sum = sum + A[i][j];
            count++;
        }
    return sum/static_cast<float>(count);
}