【入门】对角线II

输入整数N，输出相应方阵。

输入描述

一个整数N。（ 0 < n < 10 )

输出描述

一个方阵，每个数字的场宽为3。

用例输入 1 

5

用例输出 1 

  0  0  0  0  1
  0  0  0  1  0
  0  0  1  0  0
  0  1  0  0  0
  1  0  0  0  0

来源

二维数组

#include<bits/stdc++.h>
using namespace std;
int a[10][10];
int n;
int main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            if(i+j==n+1) a[i][j]=1;
            else a[i][j]=0; 
        }
    }
     for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            cout<<setw(3)<<a[i][j];
        }
        cout<<endl;
    }
    return 0;
}