更新矩阵边缘

题目要求输出矩阵，所以我使用二维数组来定义这个矩阵，每当行（列）数为1或最后一行（列）时，给里面的元素加6。最后输出即可。

#include<iostream>
using namespace std;
int main(){
    int n;
    cin>>n;
    int arr[n][n];
    for(int a=0;a<n;a++){
        for(int b=0;b<n;b++){
            cin>>arr[a][b];
            if(a==0||a==n-1||b==0||b==n-1){
                arr[a][b]=arr[a][b]+6;
            }
            printf("%3d",arr[a][b]);
            if(b==n-1){
                cout<<endl;
            }
        }
    }
    return 0;
}