『P1549』棋盘问题

题目传送门

#include<bits/stdc++.h>
using namespace std;
int n,f[15][15];
bool prime[205],use[105];
bool check(int x) {
    for(int i=2;i<=sqrt(x);i++)
     if(x % i==0) return 0;
    return 1;
}
void print() {
    for(int i=1;i<=n;i++){
        for(int j=1;j<n;j++)
         printf("%d ",f[i][j]);
        printf("%d\n",f[i][n]);
    }
    exit(0);
}

void dfs(int x,int y) { //x为行，y为列 
    if(x==n&&y==n+1) print(); 
    if(y==n+1) {dfs(x+1,1); return ;}
    if(x==1||y==1) 
     for(int i=2;i<=n*n;i++) {
         if(use[i]) continue;
         if(x==1&&!prime[f[x][y-1]+i]) continue;
         if(y==1&&!prime[f[x-1][y]+i]) continue;
         f[x][y]=i; use[i]=1;
         dfs(x,y+1);
         f[x][y]=0; use[i]=0;
     }
    else for(int i=n*n;i>=2;i--) {
        if(use[i]) continue;
        if(!prime[f[x][y-1]+i]) continue;
        if(!prime[f[x-1][y]+i]) continue;
        f[x][y]=i; use[i]=1;
        dfs(x,y+1);
        f[x][y]=0; use[i]=0;
    }
}

int main() {
    scanf("%d",&n);
    if(n==1) {printf("NO"); return 0;}
    for(int i=2;i<=200;i++) //预处理200以内的素数 
     if(check(i)) prime[i]=1;
    f[1][1]=1; use[1]=1; //左上角的格子里必须填数字1
    dfs(1,2);
    printf("NO");
}
/*
#DFS# 
题面要求：输出第一行、第一列之和为最小的排列方案。
怎样使它最小？
ANS:在第一行、第一列数字dfs的枚举时，i枚举顺序为1~n^2。(尽量取小) 
    而不是第一行、第一列的其他数字枚举时，i枚举顺序为n^2~1。(尽量取大)
PS:素数判断可以只判断左、上（已经搜索过的数）与当前和是不是素数。
*/

思路来自同校DALAO的博客。