计算n*n方阵对角线元素和

因为n未知，所以考虑使用malloc()函数，并且使用free()释放内存。

代码如下

#include 
#include 

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;
}