2016-26

/*
从键盘上输入矩阵的阶数n (n<5)，再输入矩阵的各个数据元素，
计算矩阵的对角线元素之和 sum。按照下面格式输出该矩阵和sum的值。
例如n=3时，矩阵为：
1 3 6
3 6 9
6 9 12
sum=19
*/

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

int main()
{
    int n, arr[5][5], sum = 0;
    printf("请输入阶数：\n");
    scanf("%d", &n);
    if(n > 5 || n <= 0)
    {
        printf("阶数的范围为：1-5\n");
        exit(0);
    }
    printf("请输入各个数据元素：\n");
    for(int i = 0 ; i < n ; i++)
    {
        for(int j = 0 ; j < n ; j++)
        {
            scanf("%d", &arr[i][j]);
            if(i == j)  //如果判断为对角线元素，则与sum相加
            {
                sum += arr[i][j];
            }
            printf("%d ", arr[i][j]);   //输出数组元素
        }
        printf("\n");
    }
    printf("sum = %d \n", sum);

    return 0;
}