洛谷题单P5731 蛇形方阵

题目描述
给出一个不大于 99 的正整数 n n，输出 n\times n n× n 的蛇形方阵。
从左上角填上 11 开始，顺时针方向依次填入数字，如同样例所示。注意每个数字有都会占用 33 个字符，前面使用空格补齐。
输入格式
输入一个正整数 n n，含义如题所述。
输出格式
输出符合题目要求的蛇形矩阵。
输入输出样例
输入 #1
4
输出 #1
1 2 3 4 12 13 14 5 11 16 15 6 10 9 8 7

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
const int inf=0x3f3f3f3f;
int main(){
    int n;cin>>n;
    int ans[20][20]={0};
    int i=1,k=0,r=0;
    while(i<=n*n){
        while(r<n&&!ans[k][r]) ans[k][r++]=i++;
        k++,r--;
        while(k<n&&!ans[k][r]) ans[k++][r]=i++;
        k--,r--;
        while(r>=0&&!ans[k][r]) ans[k][r--]=i++;
        k--,r++;
        while(k>=0&&!ans[k][r]) ans[k--][r]=i++;
        k++,r++;
    }
    for(i=0;i<n;i++){
     for(k=0;k<n;k++)
      printf("%3d",ans[i][k]);
     puts("");
    }
    return 0;
}

思路使用while模拟螺旋，注意数据的边界性的处理。