【模板】对角线加法-题解

描述

方馆长要对一个矩阵对角线上的数字进行操作，然后按照格式输出。

输入一个n行n列的整型矩阵，将对角线上的数加上10，再输出该矩阵。

输出每个数字占5列，右对齐。 （n<=100，数组的元素不会超过4位数）

输入描述

第一行输入一个整数n

接下来输入一个n*n的矩阵。

输出描述

输出一个n*n的矩阵。

代码

#include <bits/stdc++.h>
using namespace std;

int main(){
    int n;
    cin >> n;
    int in[n][n];
    for(int i = 0;i<n;i++){
        for(int j = 0;j<n;j++){
            cin >> in[i][j];
        }
    }
    for(int i = 0;i<n;i++){
        for(int j = 0;j<n;j++){
            if(i == j||i+j == n-1){
                in[i][j] += 10;
            }
        }
    }
    for(int i = 0;i<n;i++){
        for(int j = 0;j<n;j++){
            cout << setw(5) << in[i][j];
        }
        cout << endl;
    }
    return 0;
}