四阶幻方-蓝桥杯-DFS

答案：416

用next_permutation()全部排列的话会超时

所以用dfs搜索，只搜索前三行就好，前三行确定之后，第四行也就确定

#include<iostream>
#include<string>
#include<string.h>
#include<algorithm>
using namespace std;
int vis[],a[][];
int ans=;
int check(int x,int y)
{
    if(x<)//只枚举到前三行就可以，如果前三行符合条件，那么最后一行四个数之和一定是34
    {
        if(y<)//一行还没有放满4个，前三个数可以随便取
            return ;
        if(a[x][]+a[x][]+a[x][]+a[x][]==)
            return ;
        else
        {
            return ;
        }
    }
    else//x==3
    {
        if(y==)//判断第1列和副对角线
        {
            if((a[][]+a[][]+a[][]+a[][]!=)||(a[][]+a[][]+a[][]+a[][]!=))
                return ;
        }
        if(y==||y==)//判断第2列和第3列
            if(a[][y]+a[][y]+a[][y]+a[][y]!=)
                return ;
        if(y==)//判断第4列和主对角线
            if((a[][y]+a[][y]+a[][y]+a[][y]!=)||(a[][]+a[][]+a[][]+a[][])!=)
                return ;
        else
        {
            return ;
        }
 
    }
 
}
void dfs(int x,int y)
{
    if(x==)
    {
        ans++;
        return ;
    }
    // else if(y>3)
    //     dfs(x+1,0);
    else
    {
        for(int i=;i<=;i++)
        {
            if(vis[i]==)
            {
                vis[i]=;
                a[x][y]=i;
                if(check(x,y))
                {
                    if(y<)
                        dfs(x,y+);
                    else
                        dfs(x+,);
                }
                vis[i]=;
            }
 
        }
    }
 
}
int main()
{
    a[][]=vis[]=;
    dfs(,);
    cout<<ans<<endl;
    system("pause");
    return ;
}