利用C语言求主对角线元素之和

要求
1.二维数组输入
2.求主对角线元素之和
3.写主函数运行

主对角线特点是i和j相同

#include <stdio.h>
int main()
{
    int a[100][100];
    int n,i,j,sum=0;
    scanf("%d",&n);
    for(i=0;i<n;i++)
      for(j=0;j<n;j++)
      {
        scanf("%d",&a[i][j]);
        if(i==j)
          sum+=a[i][j];
      }
    printf("%d",sum);
    return 0;
    
}