C语言 求n阶方阵主对角线元素的和

要求：

  1）从键盘输入整数n（2<=n<=6）。

  2）从键盘输入n行n列整数，用空格分隔同一行整数。

  3）求该方阵左上到右下对角线元素之和，并输出结果（格式采用“%d”）。

运行结果示例：

输入1：

3

31 32 33

25 26 27

40 41 42

输出1：

sum=99

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

    for(i=0;i<n;i++)
    {
        for(j=0;j<n;j++)
        {
            if(i==j)
            {  
                sum=sum+a[i][i];
            }
        }
    }

    printf("sum=%d\n",sum);
    return 0;
    
}