csp PTA 程序设计综合实践4 7-6 矩阵运算 - 《C++编程基础及应用》- 习题5-8

分数 10

作者 海洋饼干叔叔

单位 重庆大学

读入一个正整数n（1≤n≤20），再读入n行n列共n²个整数并存入由二维数组表示的方阵m；计算并输出每一行的元素和；计算并输出每一列的元素和；计算并输出对角线上的元素和（即所有m[i][i]的和, i取0~n-1）。

输入样例:

3
1 2 3
4 5 6
7 8 10

输出样例:

sum of row 1 = 6.
sum of row 2 = 15.
sum of row 3 = 25.
sum of column 1 = 12.
sum of column 2 = 15.
sum of column 3 = 19.
sum of elements on the diagonal = 16.

#include<iostream>
using namespace std;

int main()
{
    int n;
    cin >> n;
    
    int r[n] = {0};  // 初始化每行的和为0
    int c[n] = {0};  // 初始化每列的和为0
    int sum = 0;     // 初始化对角线的和为0
    int a[n][n];
    
    // 输入矩阵并计算行、列、对角线的和
    for(int i = 0; i < n; i++) {
        for(int j = 0; j < n; j++) {
            cin >> a[i][j];
            r[i] += a[i][j];  // 累加第i行的元素
            c[j] += a[i][j];  // 累加第j列的元素
            if(i == j) {
                sum += a[i][j];  // 如果是对角线元素，累加对角线的和
            }
        }
    }
    
    // 输出每一行的和
    for(int i = 0; i < n; i++) {
        cout << "sum of row " << i + 1 << " = " << r[i] << "." << endl;
    }
    
    // 输出每一列的和
    for(int j = 0; j < n; j++) {
        cout << "sum of column " << j + 1 << " = " << c[j] << "." << endl;
    }
    
    // 输出对角线的和
    cout << "sum of elements on the diagonal = " << sum << "." << endl;
    
    return 0;
}