Функция находящая сумму элементов главной диагонали матрицы

#include <iostream>
using namespace std;
 
int diagonal(char c) {
    int s, i, j;
    s = 0;
    for (i=0; i<N; i++)
         s += a[i][i];
    return s;    
}
 
int main()
{
   setlocale(LC_ALL, "Russian");
   int n,m, s, i, j,sum;
   char ch;
   cout<<"Введите размер матрицы А:";
   cin>>n;
   cout<<"Введите размер матрицы В:";
   cin>>m;
   int **arr=new int*[n];
   int **ar=new int*[m];
   cout<<"Matrix A"<<"\n";
   for (int i=0; i<n;i++)
   {
       arr[i]= new int[n];
   }
   for (int i=0; i<n;i++)
   {
       for (int j=0;j<n;j++)
       {
           arr[i][j]=rand()%20;
           cout<<arr[i][j]<<"\t";
       }
       cout<<endl;
   }
   cout<<"Matrix B"<<"\n";
   for (int i=0; i<m;i++)
   {
       ar[i]= new int[m];
   }
   for (int i=0; i<m;i++)
   {
       for (int j=0;j<m;j++)
       {
           ar[i][j]=rand()%20;
           cout<<ar[i][j]<<"\t";
       }
       cout<<endl;
   }
   return 0;
}

int diagonal(int **a,int n) 
{
    int s=0;
    for (i=0; i<n; i++)
         s += a[i][i];
    return s;    
}
.........
int sa=diagonal(arr,n);
int sb=diagonal(ar,m);