AcWing 754. 平方矩阵 II

输入整数N，输出一个N阶的二维数组。

数组的形式参照样例。

输入格式

输入包含多行，每行包含一个整数N。

当输入行为N=0时，表示输入结束，且该行无需作任何处理。

输出格式

对于每个输入整数N，输出一个满足要求的N阶二维数组。

每个数组占N行，每行包含N个用空格隔开的整数。

每个数组输出完毕后，输出一个空行。

数据范围

0≤N≤100

扫描二维码关注公众号，回复： 12268012 查看本文章

输入样例：

1
2
3
4
5
0

输出样例：

1

1 2
2 1

1 2 3
2 1 2
3 2 1

1 2 3 4
2 1 2 3
3 2 1 2
4 3 2 1

1 2 3 4 5
2 1 2 3 4
3 2 1 2 3
4 3 2 1 2
5 4 3 2 1

思路： 填入上三角部分的同时将值填入对应下三角部分即ans[j][i]=ans[i][j]

#include<iostream>
#include<vector>
using namespace std;
int n,k=1;
int main(){
    while(cin>>n && n!=0){
        vector<vector<int> > ans(n,vector<int>(n,0));
        for(int i=0;i<n;i++){
            k=1;
            for(int j=0;j<n;j++){
                if(i<=j) {ans[i][j]=k++;ans[j][i]=ans[i][j];}
            }
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
               cout<<ans[i][j]<<' ';
            }
            cout<<endl;
        }
        cout<<endl;
    }
    return 0;
}