首页 > > 程序设计 > C/C++ >

spiral matrix

2018-06-17 23:50:45来源：未知阅读 ()

如图，生成如此旋转的矩阵输入行N，列M，求上述矩阵

假定纵向为x轴，横向为y轴，从1开始

层数 k=min（i,j,N+1-i,M+1-j）

k层矩阵的长 l=N-(k-1)*2 宽 w=M-(k-1)*2

矩阵左上角的坐标值为（k,k）,根据外层矩阵的长宽可以算出数值，进而确定当前矩阵（i,j）的值

缺点：代码中有很多重复计算，可以通过记录一些值来减少计算次数

#include<stdio.h>

#define min(x,y) (x)>(y)?(y):(x)
int N,M;
int foo(int,int);

int main()
{

    while(scanf("%d %d",&N,&M)==2)
        for(int i=1;i<=N;i++)
        {
            for(int j=1;j<=M;j++)
            {
                printf("%3d ",foo(i,j));
            }
            printf("\n");
        }
}



int foo(int i,int j)
{
    int k=min( min(i,j) , min(N+1-i,M+1-j) );
    int l=N-(k-1)*2, w=M-(k-1)*2;
    int s=0;
    
    for( int t=1;t<k;t++ )
    {
        int l=N-(t-1)*2, w=M-(t-1)*2;
        s+=(l-1)*2+(w-1)*2;
    }
    s++;
    
    if(l==1) s+=j-k;
    else if(w==1) s+=i-k;
    else
    {
        if(i==k) s+=j-k;
        else if(j==k+w-1) s+=w-1+i-k;
        else if(i==k+l-1) s+=w-1+l-1+k+w-1-j;
        else s+=w-1+l-1+w-1+k+l-1-i;
    }
    return s;
}