因为n未知,所以考虑使用malloc()函数,并且使用free()释放内存。
代码如下
#include <stdio.h>
#include <stdlib.h>
void ReadMatrix(int** a, int n);
int AddMatrix(int** a, int n);
int main(void)
{
int n,sum;
int i, j;
int** pa;
printf("请输入方阵的阶数N:");
scanf("%d", &n);
pa = (int**)malloc(sizeof(int*) * n);//使用malloc申请内存存放n阶方阵
for (i = 0; i < n; i++) {
pa[i] = (int*)malloc(sizeof(int) * n);
}
for (i = 0; i < n; i++) {//初始化数组,元素全为0;
for (j = 0; j < n; j++) {
pa[i][j] = 0;
}
}
printf("请输入方阵(同行内每个元素之间用空格隔开,每一行输入完毕用回车键进入下一行输入):\n");
ReadMatrix(pa, n);//给数组赋值
sum = AddMatrix(pa, n);
printf("两条对角线的和是:%d", sum);
for (i = 0; i < n; i++) {//释放内存
free(pa[i]);
}
free(pa);
return 0;
}
void ReadMatrix(int** a,int n)
{
int i, j;
for (i = 0; i < n; i++) {
for (j = 0; j < n; j++) {
scanf("%d", a[i]+j);
}
}
return;
}
int AddMatrix(int** a, int n)
{
int i, sum = 0;
int nRow = 0, nCol = 0;
for (i = 0,sum = 0; i < n; i++) {
sum += a[nRow][nCol];
nRow++, nCol++;
}
nRow = n - 1,nCol = 0;
for (i = 0; i < n; i++) {
sum += a[nRow][nCol];
nRow--, nCol++;
}
return sum;
}