题解 | 蛇形矩阵

#include<iostream>
using namespace std;
int main()
{
    int n;
    int arr[1001][1001];
    cin>>n;
    int k=0;
    int i=0,j=0;
    arr[i][j]=++k;
    while(k<n*n)
    {
       while(i-1>-1&&j+1<n)
       {
         i--,j++;
         arr[i][j]=++k;   
       }
       if(j+1<n)
       {
           j++;
       arr[i][j]=++k;
       }
       else 
       {
           i++;
           arr[i][j]=++k;
       }
       while(i+1<n&&j-1>-1)
       {
        i++,j--;
        arr[i][j]=++k;
       }
       if(i+1<n)
       {
        i++;
        arr[i][j]=++k;
       }
       else 
       {
          j++;
          arr[i][j]=++k;       
       }
    }
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            cout<<arr[i][j]<<" ";
        }
        cout<<endl;
    }
    return 0;
}