蓝桥杯 四阶幻方

把1~16的数字填入4x4的方格中，使得行、列以及两个对角线的和都相等，满足这样的特征时称为：四阶幻方。

四阶幻方可能有很多方案。如果固定左上角为1，请计算一共有多少种方案。
比如：
1 2 15 16
12 14 3 5
13 7 10 4
8 11 6 9

以及：
1 12 13 8
2 14 7 11
15 3 10 6
16 5 4 9

就可以算为两种不同的方案。

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;
int num[5][5],vis[20];
int cnt=0;
int check1(int cur)
{   //行检测
    if  (num[cur/4][0]+num[cur/4][1]+num[cur/4][2]+num[cur/4][3]==34)
        return 1;
    return 0;
}
int check2(int cur)
{   //列检测
    if (num[0][cur%4]+num[1][cur%4]+num[2][cur%4]+num[3][cur%4]==34)
        return 1;
    return 0;
}
int check3()
{   //斜边检测
    int i,sum1=0,sum2=0;
    for (i=0;i<4;i++)
        sum1+=num[i][i],sum2+=num[4-i-1][i];
    if (sum1==sum2&&sum2==34)
        return 1;
    return 0;
}
void dfs(int cur)
{
    int i,j;
    if (cur%4==0)
        if (!check1(cur-1))
            return;
    if (cur>=13)
        if (!check2(cur-1))
            return;
    if (cur==16)
        if (check3())
    {
        for (i=0;i<4;i++)
        {
        for (j=0;j<4;j++)
            printf("%d ",num[i][j]);
        printf("\n");
        }
        printf("\n");
         cnt++;
    }
    for (i=1;i<=16;i++)
        if (!vis[i])
    {
        vis[i]=1;
        num[cur/4][cur%4]=i;
        dfs(cur+1);
        vis[i]=0;
    }
    return;
}
int main()
{
    memset(vis,0,sizeof(vis));
    num[0][0]=1;
    vis[1]=1;
    dfs(1);
    printf("%d\n",cnt);
    return 0;
}