Задача на подсчет суммы главной диагонали

#include <math.h>
#include <iostream>
#include <iomanip>
#include <locale.h>
using namespace std;
 
void main();
void rez(double**,int,int);     //расчет согласно варианта
void arr(int,int);      //выделение памяти
double in(double**,int,int,int ,double);    //заполнение массива
void out(double**,int,int,int,int);     //вывод массива на основе потока cout
void output(double**,int,int,int,int);  //вывод массива с использованием printf()
double lod(int,int,int,double);     //расчет значения элемента массива
int mem;    //для определения способа выделения памяти
double **a;     //объявление массива
 
void main()
{
    int num;    //для определения способа вывода массива
    int x,y,precision,width;    //x-кол-во строк,y-кол-во столбцов,precision-точность вещественного числа,width-ширина столбца
    double v;
    int b;
    setlocale(LC_ALL,"Russian");
    cout<<"Введите кол-во строк: ";
    cin>>x;
    cout<<"Введите кол-во столбцов: ";
    cin>>y;
    cout<<"Введите вещественное число p: ";
    cin>>v;
    cout<<"Введиет целое число B: ";
    cin>>b;
    arr(x,y);   //вызов функции
    in(a,x,y,b,v);
    cout<<"Выберите способ вывода массива printf() или cout (0/1): ";
    cin>>num;
    cout<<"Введите ширину столбца: ";
    cin>>width;
    cout<<"Введите точность для вещественых полей: ";
    cin>>precision;
    if(num==0)
    {
        output(a,x,y,width,precision);
        out(a,x,y,width,precision);
    }
    else if(num==1)
    {
        out(a,x,y,width,precision);
        cout<<endl;
        output(a,x,y,width,precision);
    }
    rez(a,x,y);
    if(mem==1||2)   //если память выделялась используя malloc() или calloc()
        free(a);
    else if(mem==0)     //если память выделялась с помощью new()
        delete []a;
    system("pause");
    return;
}
void rez(double **a, int x,int y)
 {
double n;
int i;
int h=0;
   for(int i=0; i<n; i++)
   {
    h=h+a[i][i];
   }
cout<<"сумма: "<<h<<endl;
 
 }
void arr(int x,int y)
{
    cout<<"Выделить память с помощью new(), malloc() или calloc()? (0/1/2): ";
    cin>>mem;
    if(mem==0)
    {
        a=new double *[x];      //динамическое выделение памяти под массив
        for(int i=0;i<x;i++)
            a[i]=new double [y];
    }
    else if(mem==1)
    {
        a=(double **)malloc(x*sizeof(double*));
        for(int i=0;i<x;i++)
        {
            a[i]=(double *)malloc(y*sizeof(double));
        }
    }
    else if(mem==2)
    {
        a=(double **)calloc(x,sizeof(double*));
        for(int i=0;i<x;i++)
        {
            a[i]=(double *)calloc(y,sizeof(double));
        }
    }
    else
    {
        exit(1);
    }
}
double lod(int x,int y,int b,double z)
{
    double p;
    double lowerPart;
lowerPart = b * y - exp(-x) * sin(abs(x/(b+1)) ) + 1;
    double upperPart;
    if(! lowerPart) lowerPart++;
upperPart = z*z * x + exp(-y) * pow( cos( x / b ), 2.0 );
p = lowerPart / upperPart;
 
    return p;
}
double in(double **a,int x,int y,int b,double v)
{
    for(int i=0;i<x;i++)
    {
        for(int j=0;j<y;j++)
        {
            a[i][j]=lod(i,j,b,v);   //заполнение массива значениями расчитаными по формуле, данной в задании
        }
    }
    return **a;
}
void out(double **a,int x,int y,int width,int precision)
{
    for(int i=0;i<x;i++)
    {
        cout<<endl;
        for(int j=0;j<y;j++)
            cout<<setw(width)<<fixed<<setprecision(precision)<<a[i][j];     //setw()-ширина столбца, setprecision()-точность вещественного числа
    }
        cout<<endl;
}
void output(double **a,int x,int y,int width,int precision)
{
    for(int i=0;i<x;i++)
    {
        for(int j=0;j<y;j++)
            printf("%*.*f", width, precision, a[i][j]);
        printf("\n");
    }
}

void rez(double **a, int x,int y)
{
    int n;
    n=x<y?x:y;
    double h=0;
    for(int i=0; i<n; i++)
    {
        h+=a[i][i];
    }
    cout<<"сумма: "<<h<<endl;
 
 }

double lod(int x,int y,int b,double z)
{
    double p;
    double lowerPart;
    lowerPart = b * y - exp(-(double)x) * sin(fabs((double)x/(b+1.0)) ) + 1.0;
    double upperPart;
    if(! lowerPart) lowerPart++;
    upperPart = z*z * x + exp(-(double)y) * pow( cos( (double)x / b ), 2 );
    p = lowerPart / upperPart;
 
    return p;
}