3070 B20 例题-7 矩阵对角线

题目描述

在一个二维的数字矩阵中，从左上角至右下角的对角线为主对角线，从右上角至左下角的对角线为次对角线，如下图所示。

已知一个 n×n 的数字矩阵，请你输出矩阵主，次对角线上的元素。

输入格式

一行一个整数 n ；

接下来的 n 行每行 n 个整数 ai,j​，表示矩阵元素的值。

输出格式

两行：

第一行 n 个数，依次输出主对角线上的元素（从左上角至右下角）；

第二行 n 个数，依次输出次对角线上的元素（从右上角至左下角）；

每个输出元素占宽为 55 且右对齐输出。

样例输入

3
1 2 3
1 2 3
1 2 3

样例输出

    1    2    3
    3    2    1

数据范围

对于 100%100% 的数据：1≤n≤100,−100≤ai,j​≤100 。

代码：

#include<bits/stdc++.h>
using namespace std;
int a[107][107];
int b[107];
int main(){
    int n;
    cin>>n;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            cin>>a[i][j];
        }
    }
    for(int i=1;i<=n;i++){
        printf("%5d",a[i][i]); 
    }
    cout<<endl;
    for(int i=1;i<=n;i++){
        printf("%5d",a[i][n-i+1]); 
    }
}