螺旋矩阵

Posted 2020-11-22 chilkings

输出一个n*m的螺旋矩阵

#include<stdio.h>
#include<iostream>
using namespace std;

int dir[4][2] = {{1,0},{0,1},{-1,0},{0,-1}};

int a[1005][1005];

int n,m;
int x,y;
int k=2;
int i=0;

void dfs(int x,int y)
{
    // if(k>n*m) return;
    while(true)
    {
        int nx=x+dir[i][1];
        int ny=y+dir[i][0];
        if(0<=nx && nx<n && 0<=ny && ny<m && a[nx][ny]==0)
        {
            a[nx][ny]=k;
            k++;
            dfs(nx,ny);
            break;
        }
        else{
            i=(i+1)%4;
            if(k > n * m) break;
        }
    }
    return ;
}


int main()
{
    scanf("%d%d",&n,&m);
    a[0][0]=1;
    dfs(0,0);
    for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)
            {
                printf("%4d ",a[i][j]);
            }
            printf("
");
        }
    return 0;
}