编写程序，求某一个方阵中对角线元素和，要求，考虑主对角和副对角线，c语言代码简洁

以下是C语言代码，可以求解任意大小的方阵中主对角线和和副对角线和：

#include <stdio.h>

int main() {
    int n, i, j, sum1 = 0, sum2 = 0;
    printf("请输入方阵的大小：");
    scanf("%d", &n);
    int a[n][n];
    printf("请输入方阵的元素：\n");
    for (i = 0; i < n; i++) {
        for (j = 0; j < n; j++) {
            scanf("%d", &a[i][j]);
            if (i == j) {  // 计算主对角线和
                sum1 += a[i][j];
            }
            if (i + j == n - 1) {  // 计算副对角线和
                sum2 += a[i][j];
            }
        }
    }
    printf("主对角线和为：%d\n", sum1);
    printf("副对角线和为：%d\n", sum2);
    return 0;
}