matriz identidade ou não

#include <stdio.h>
#include <stdlib.h>

int matriz_identidade (int n, int **mat);

int main (void)

{
int **matriz;
int i, j, n, flag;


printf ("Digite a dimensão da matriz quadrada\n");
scanf ("%d", &n);

printf ("Alocação memoria para a matriz\n");
matriz = (int **) malloc(n* sizeof(int*));

for (i=0; i<n; i++)
{
matriz[i]= (int *) malloc(n*sizeof(int));

}

printf ("Leitura da matriz\n");

for (i=0; i<n; i++)
{
for (j=0; j<n; j++)
{
printf ("Digite valor\n");
scanf ("%d", &matriz[i][j]);

}

}

flag = matriz_identidade (n, matriz);

if (flag == 1)
{
printf ("Matriz identidade\n");
}

else
{
printf ("A matriz nao eh identidade\n");
}

system ("pause");
return 0;

}


int matriz_identidade (int n, int **mat)

{
int i,j;

for (i=0; i<n; i++)
{
for (j=0; j<n; j++)
{
if (i==j)
{
if (mat[i][j] !=1)
{

return 0;
}
}


else
{
if (mat[i][j] != 0)
{
return 0;
}
}
}
}
return 1;