输出矩阵两个对角线上的数(二维数组)

编写程序，将数据输入一个n * n二维数组中，输出矩阵两个对角线上的数。假设数据为整型数据。

输入格式:

第1行，输入 n ，表示矩阵的行列数。n在2-100之间。

第2行到n+1行，输入n行数据，每行n个数据，每行的n个数据之间用一个空格分隔。

输出格式:

输出二行，矩阵两个对角线上的数。第一行为从左上到右下对角线，第二行从左下到右上对角线上的值。。

输入样例:

在这里给出一组输入。例如：

4
25 36 78 13
12 26 88 93
75 18 22 32
56 44 36 58

输出样例:

在这里给出相应的输出。例如：

25 26 22 58
56 18 88 13

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

个人萌新代码，仅供参考

#include<iostream>
using namespace std;
int main()
{
    int a;
    cin>>a;
    int b[a][a];
    for(int i=0;i<a;i++)
    {
        for(int j=0;j<a;j++)
        {
            cin>>b[i][j];
        }
    }
    for(int i=0;i<a-1;i++)
    {
        cout<<b[i][i]<<" ";
    }
    
    cout<<b[a-1][a-1]<<endl;
    int c=a-1;
    for(int i=0;i<a-1;i++)
    {
        cout<<b[c][i]<<" ";
        c--;
    }
    cout<<b[0][a-1]<<endl;
    return 0;
}