洛谷——P1549 棋盘问题（2）

P1549 棋盘问题（2）

搜索||打表

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cstring>
#include<cmath>
#include<cstdlib>

using namespace std;

int n,p[][],an[][],hang,lie,ans[][];
bool vi[];

void print(){
    for(int i=;i<=n;i++){
        for(int j=;j<=n;j++)
            printf("%d ",ans[i][j]);
        puts("");
    }
}

void scz(){
    int han=,li=;
    for(int i=;i<=n;i++) han+=an[][i],li+=an[i][];
    if(han>hang||li>lie) return;
    hang=han,lie=li;
    for(int i=;i<=n;i++)
        for(int j=;j<=n;j++)
            ans[i][j]=an[i][j];
}

void dfs(int x,int y){
    if(y==n+){
        int sum=;
        if(x==&&hang!=) {
            for(int i=;i<=n;i++) sum+=an[][i];
            if(sum>hang) return;
            hang=sum;
        }
        x++,y=;
    }
    if(x==n+&&y==) {
        scz();
        print();
        exit();
        return;
    }
    if(y==&&x==n&&lie!=){
        int sum=;
        for(int i=;i<=n;i++) sum+=an[i][];
        if(sum>lie) return;
        lie=sum;
    }
    if(x==||y==){
        for(int i=;i<=n*n;i++){
            if(vi[i]) continue;
            int z=an[x][y-],s=an[x-][y];
            if(z&&!p[z][i]) continue;
            if(s&&!p[s][i]) continue;
            an[x][y]=i;vi[i]=;
            dfs(x,y+);
            an[x][y]=;vi[i]=;
        }
    }
    else {
        for(int i=n*n;i>=;i--){
            if(vi[i]) continue;
            int z=an[x][y-],s=an[x-][y];
            if(z&&!p[z][i]) continue;
            if(s&&!p[s][i]) continue;
            an[x][y]=i;vi[i]=;
            dfs(x,y+);
            an[x][y]=;vi[i]=;
        }
    }
}

bool tp(int x){
    for(int i=;i<=sqrt(x);i++)
        if(x%i==) return false;
    return true;
}

void yu(){
    for(int i=;i<=n*n;i++){
        for(int j=;j<=n*n;j++){
            if(tp(i+j)) p[i][j]=;
　　}
    }
}

int main()
{
    scanf("%d",&n);
    if(n==) printf("NO\n");
    else{
        hang=lie=;
        yu();
        vi[]=;
        an[][]=;
        dfs(,);
        if(hang==) cout<<"NO\n";
        else print();
    }
    return ;
}