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

#include<iostream.h>
#include<math.h>
void main()
{
int i,j,c;
int a[3][3];
for(i = 0; i < 3; i++)
{
for(j = 0; j < 3; j++)
{
cout<<"vvedite massiv a["<<i<<"]["<<j<<"] =";
cin>>a[i][j];
}}
cout << endl;
int b[3]={5,3,7};
for(c=0;c<3;c++)
{
cout<< b[c]<<" ";
}
cout<<endl;
cout<<endl;
c=0;
for(i = 0; i < 3; i++)
{
for(j = 0; j < 3; j++)
{
if(i==j)
{ 
a[i][j]=b[c];
c++;
}
}
}
for(i = 0; i < 3; i++)
{
for(j = 0; j < 3; j++)
{
cout << a[i][j] << " ";
}       
cout << endl;
}
}

#include <iostream>
#include <iomanip>
#include <cstdlib>
#include <ctime>
 
const int matrixSize = 3;
 
void function (int (*matrix)[matrixSize], int (*array));
 
int main ()
{
    srand(time(0));
    int matrix[matrixSize][matrixSize];
    int array[matrixSize] = {5, 3, 7};
 
    std::cout << "Matrix: " << std::endl;
    for (int i = 0; i < matrixSize; i++) {
        for (int j = 0; j < matrixSize; j++)
            std::cout << std::setw(5) << (matrix[i][j] = rand () % 25);
        std::cout << std::endl;
    }
 
    function(matrix, array);
    std::cout << "Matrix: " << std::endl;
    for (int i = 0; i < matrixSize; i++) {
        for (int j = 0; j < matrixSize; j++)
            std::cout << std::setw(5) << matrix[i][j];
        std::cout << std::endl;
    }
 
    return 0;
}
 
void function (int (*matrix)[matrixSize], int (*array))
{
    for (int i = 0; i < matrixSize; i++)
        matrix[i][i] = array[i];
}