Viết hàm đếm số lượng số nguyên tố trong ma trận vuông các số nguyên?

#include<stdio.h>

#include<conio.h>

#include<math.h>

#define MAX 100

void NhapMaTran(int a[][MAX], int &n)

{

do

{

printf("\nNhap n: ");

scanf("%d", &n);

if(n < 1 || n > MAX)

{

printf("\nSo phan tu khong hop le. Xin kiem tra lai !");

}

}while(n < 1 || n > MAX);

for(int i = 0; i < n; i++)

{

for(int j = 0; j < n; j++)

{

printf("\nNhap vao a[%d][%d] = ", i, j);

scanf("%d", &a[i][j]);

}

}

}

void XuatMaTran(int a[][MAX], int n)

{

for(int i = 0; i < n; i++)

{

for(int j = 0; j < n; j++)

{

printf("%4d",a[i][j]);

}

printf("\n\n");

}

}

int KiemTraNguyenTo(int n)

{

if(n < 2)

{

return 0;

}

else if(n > 2)

{

if(n % 2 == 0)

{

return 0;

}

for(int i = 3; i <= sqrt((float)n); i += 2)

{

if(n % i == 0)

{

return 0;

}

}

}

return 1;

}

int DemSoLuongSoNguyenTo(int a[][MAX], int n)

{

int dem = 0;

for(int i = 0; i < n; i++)

{

for(int j = 0; j < n; j++)

{

if(KiemTraNguyenTo(a[i][j]) == 1)

{

dem++;

}

}

}

return dem;

}

int main()

{

int a[MAX][MAX], n;

NhapMaTran(a,n);

XuatMaTran(a,n);

int dem = DemSoLuongSoNguyenTo(a, n);

printf("\nSo luong so nguyen to trong ma tran vuong = %d", dem);

getch();

return 0;

}