P2615 [NOIP2015 提高组] 神奇的幻方

//P2615

#include<stdio.h>

int main()

{

    int N[40][40]={0};

    int n,k,row,col;

    scanf("%d",&n);

    N[0][n/2]=1;row=0;col=n/2;

        for(k=2;k<=n*n;k++)

        {

            if(row==0&&col!=n-1){N[n-1][col+1]=k;row=n-1;col=col+1;}

            else if(col==n-1&&row!=0){N[row-1][0]=k;row=row-1;col=0;}

            else if(row==0&&col==n-1){N[row+1][col]=k;row=row+1;}

            else if(row!=0&&col!=n-1)

            {

                if(N[row-1][col+1]==0){N[row-1][col+1]=k;row=row-1;col=col+1;}

                else {N[row+1][col]=k;row=row+1;}

            }

        }

    for(int i=0;i<n;i++)

    {

        for(int j=0;j<n;j++)

        {

            printf("%d ",N[i][j]);

        }

        printf("\n");

    }

    return 0;

}