棋盘问题(NOIP提高组1997)

传送门
相当古老的题了
一年前做这一题一直WA,今天简单看看就AC
感觉自己水平有所提高
没有,没有,还是很菜的
这道题洛谷给的标签高了,最多也就普及+
下面步入正题
数据水得惊人,直接dfs即可
但是我们要注意dfs的顺序,因为它有一个要求,就是第一行和第一列之和最小
所以我们的顺序就是:先填第一行,再填第一列,然后把剩下的小矩阵按一行行从左到右的顺序填上即可
代码如下:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int ans[10][10];
int vis[101];
int n;
inline bool isprime(int x){
    int q=sqrt(x+0.5);
    for(int i=2;i<=q;i++){
        if(x%i==0){
            return false;
        }
    }
    return true;
}
inline bool dfs(int dep,int i,int j,int ok){
    if(dep>n*n){
        return true;   //递归边界
    }
    for(int a=2;a<=n*n;a++){
        if(vis[a]==0){
            if(i!=0&&!isprime(a+ans[i-1][j])){
                continue;
            }
            if(j!=0&&!isprime(a+ans[i][j-1])){
                continue;
            }
            vis[a]=1;
            ans[i][j]=a;
            int ii,jj;
            int fla=ok;//如果直接改动ok,会导致返回时出现错误
            if(ok){     //控制遍历顺序
                ii=i,jj=j+1;
                if(jj>=n){
                    jj=1;
                    ii++;
                }
            }else{
                ii=i,jj=j;
                if(ii==0){
                    if((jj+1)>=n){
                        ii=1,jj=0;
                    }else{
                        jj++;
                    }
                }else{
                    if((ii+1)>=n){
                        ii=1,jj=1;
                        fla=1;
                    }else{
                        ii++;
                    }
                }
            }

            if(dfs(dep+1,ii,jj,fla)){
                return true;
            }else{
                vis[a]=0;
                ans[i][j]=0;
            }
        }
    }

    return false;
}
int main(){
    scanf("%d",&n);
    memset(vis,0,sizeof(vis));
    memset(ans,0,sizeof(ans));
    vis[1]=1;
    ans[0][0]=1;
    if(n!=1&&dfs(2,0,1,0)){
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                printf("%d",ans[i][j]);
                if(j!=n-1){
                    printf(" ");
                }
            }
            if(i!=n-1){
                printf("\n");
            }
        }
    }else{
        printf("NO");
    }
    return 0;
}  

猜你喜欢

转载自blog.csdn.net/cggwz/article/details/81208592