判断矩阵是否为单位矩阵

/**
*单位矩阵就是一个正方形矩阵，它除了主对角线的元素值为1以后，其余元素的值均为0
*/
include <stdio.h> 
 
#define TRUE  1 
#define FALSE 0 
 
int is_identity_matrix( int matrix[10][10] ); 
 
int main() 
{ 
    int result; 
    int mat[10][10] = { 
                        {1,0,0,0,0,0,0,0,0,0}, 
                        {0,1,0,0,0,0,0,0,0,0}, 
                        {0,0,1,0,0,0,0,0,0,0}, 
                        {0,0,0,1,0,0,0,0,0,0}, 
                        {0,0,0,0,1,0,0,0,0,0}, 
                        {0,0,0,0,0,1,0,0,0,0}, 
                        {0,0,0,0,0,0,1,0,0,0}, 
                        {0,0,0,0,0,0,0,1,0,0}, 
                        {0,0,0,0,0,0,0,0,1,0}, 
                        {0,0,0,0,0,0,0,0,0,1} 
                      }; 
 
    result = is_identity_matrix( mat ); 
 
    if( result ) 
        printf("mat is identity matrix\n"); 
    else 
        printf("mat is not identity matrix\n"); 
 
    return 1; 
 
} 
 
int is_identity_matrix( int matrix[10][10]) 
{ 
    int  i , j ; 
    int flags = 1; 
    for( i = 0 ; i < 10 ; i++ ) 
    { 
        for( j = 0 ; j < 10 ; j++ ){ 
            if( matrix[i][i] != 1 ){ 
                flags = 0;   
            } 
            if( i != j && matrix[i][j] != 0){ 
                flags = 0; 
            } 
        } 
    } 
    if( flags == 0 ) 
        return FALSE; 
    else 
        return TRUE; 
} 

 

vim 7.3

gcc 版本 : 4.5.1 20100924 (Red Hat 4.5.1-4) (GCC)  
system : Fedora 14