PTA7-2 方阵边上元素求和

输入格式:

输入数据的第一行为一个正整数T，表示测试数据的组数。然后是T组测试数据。
每组测试数据的第一行为两个整数n和k(3≤n≤15，1≤k<n/2)，n表示方阵的阶数，k表示方阵从外到内四周的圈数。然后输入n行，每行n个整数。

输出格式:

对于每组测试，输出的形式为“sum=num”，其中num为所求的和值（结果保证小于231）。

 输入样例: 

2
3 1
9 8 7
3 4 8
7 5 6
5 2
9 2 5 7 2
2 6 7 2 7
2 8 5 1 4
9 6 8 7 7
3 7 9 3 8
 

输出样例：

 sum=53
sum=131

代码展示：

#include <stdio.h>

#define MAX_N 15//限制长度

int main() {
    int t, n, k;
    scanf("%d", &t); // 读取测试数据的组数

    while (t--) {
        scanf("%d %d", &n, &k); // 读取方阵的阶数和圈数
        int a[MAX_N][MAX_N];//读取数组
        int sum = 0;

        // 读取方阵中的所有元素
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                scanf("%d", &a[i][j]);
            }
        }

        // 计算四周k圈上的数字之和
        for (int layer = 0; layer < k; layer++) {
            int rowStart = layer;
            int rowEnd = n - 1 - layer;
            int colStart = layer;
            int colEnd = n - 1 - layer;

            for (int i = rowStart; i <= rowEnd; i++) {
                sum += a[i][colStart];
                sum += a[i][colEnd];
            }
            for (int j = colStart + 1; j < colEnd; j++) {
                sum += a[rowStart][j];
                sum += a[rowEnd][j];
            }
        }

        // 输出结果
        printf("sum=%d\n", sum);
    }

    return 0;
}