C语言训练-求矩阵的两对角线上的元素之和(C语言）

参考代码:

#include<stdio.h> 

int main ()

{

     int i,j,N,sum=0;

     scanf("%d",&N);

     int arr[N][N];

     for(i=0;i<N;i++)

          for(j=0;j<N;j++)

            {

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

                 if(i==j||i==N-j-1)

                     sum+=arr[i][j];

             }

     printf("%d",sum);

     return 0;

}