题面
Description
在下面的方格中，每行，每列，以及两条对角线上的数字可以看作是五位的素数。方格中的行按照从左到右的顺序组成一个素数，而列按照从上到下的顺序。两条对角线也是按照从左到右的顺序来组成。
这些素数各个数位上的和必须相等。
左上角的数字是预先定好的。
一个素数可能在方阵中重复多次。
如果不只有一个解，将它们全部输出（按照这25个数字组成的25位数的大小排序）。
一个五位的素数开头不能为0（例如：00003 不是五位素数）
Input
一行包括两个被空格分开的整数:各个位的数字和 和左上角的数字。
Output
对于每一个找到的方案输出5行，每行5个字符, 每行可以转化为一个5位的质数.在两组方案中间输出一个空行. 如果没有解就单独输出一行"NONE"。
Sample Input
11 1
Sample Output
11351
14033
30323
53201
13313
11351
33203
30323
14033
33311
13313
13043
32303
50231
13331
题解
这道题，搜索好题。
先说我的第一种方法：
首先筛选质数（一个很显然的优化：只需要数字和为给定值的质数）
然后是将质数建立字典树
然后从第一列开始，每列填写一个质数，然后每行每对角线依次检查，判断是否可行。每次的解先存起来，最后排序、输出。
长长的代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<vector>
#include<algorithm>
using namespace std;
bool zs(int x)
{
int t=sqrt(x);
for(int i=2;i<=t;++i)
if(x%i==0)return false;
return true;
}
int cnt=0,tot=0,pri[5000];
int S,A;
int g[6][6];
int number[5000][6];
struct Node
{
int vis[10];//字典树
}t[100000];
struct Ans
{
int g[6][6];
}ans[100];
bool operator <(Ans a,Ans b)
{
for(int i=1;i<=5;++i)
for(int j=1;j<=5;++j)
if(a.g[i][j]!=b.g[i][j])
return a.g[i][j]<b.g[i][j];
}
int sum=0;
inline void outp()
{
for(int i=1;i<=5;++i)
{
for(int j=1;j<=5;++j)
printf("%d",g[i][j]);
printf("
");
}
}
inline void Record()//记录答案
{
sum++;
for(int i=1;i<=5;++i)
for(int j=1;j<=5;++j)
ans[sum].g[i][j]=g[i][j];
}
inline void Add(int x)//判断是否可行并构建字典树
{
int a[6]={0,x/10000,x/1000%10,x/100%10,x/10%10,x%10};
if(a[1]+a[2]+a[3]+a[4]+a[5]!=S)return;
if(!zs(x))return;
int now=0;
++tot;//满足条件的素数
pri[tot]=x;
//number[tot]=a;
//cout<<pri[tot]<<endl;
for(int i=1;i<=5;++i)//构造字典树
{
if(t[now].vis[a[i]]==0)
t[now].vis[a[i]]=++cnt;
now=t[now].vis[a[i]];
}
}
inline bool check(int x)//检查到第x列
{
int now,s;
//检验左上到右下对角线
s=now=0;
for(int j=1;j<=x;++j)
{
s+=g[j][j];
if(t[now].vis[g[j][j]]==0)return false;
now=t[now].vis[g[j][j]];
}
if(s>S)return false;//如果和已经大于S
if(x==5&&s!=S)return false;//最后的和必须为S
//检验左下到右上对角线
s=now=0;
for(int j=1;j<=x;++j)
{
s+=g[6-j][j];
if(t[now].vis[g[6-j][j]]==0)return false;
now=t[now].vis[g[6-j][j]];
}
if(s>S)return false;//如果和已经大于S
if(x==5&&s!=S)return false;//最后的和必须为S
for(int i=1;i<=5;++i)//枚举行
{
now=s=0;
for(int j=1;j<=x;++j)
{
s+=g[i][j];
if(t[now].vis[g[i][j]]==0)return false;
now=t[now].vis[g[i][j]];//字典树
}
if(s>S)return false;//如果和已经大于S
if(x==5&&s!=S)return false;//最后的和必须为S
}
return true;
}
void DFS(int x)
{
//outp();
//cout<<endl;
if(!check(x-1))return;//不合要求
if(x==6)
{
Record();
//outp();
return;
}
for(int i=1;i<=tot;++i)//填数
{
g[1][x]=pri[i]/10000;
g[2][x]=pri[i]/1000%10;
g[3][x]=pri[i]/100%10;
g[4][x]=pri[i]/10%10;
g[5][x]=pri[i]%10;
DFS(x+1);
g[1][x]=g[2][x]=g[3][x]=g[4][x]=g[5][x]=-1;//回朔
}
}
int main()
{
freopen("prime.in","r",stdin);
freopen("prime.out","w",stdout);
cin>>S;
cin>>A;
for(int i=10003;i<=99999;i+=2)
Add(i);
memset(g,-1,sizeof(g));
for(int i=1;i<=tot;++i)//枚举第一列
{
if(pri[i]/10000==A)
{
g[1][1]=A;
g[2][1]=pri[i]/1000%10;
g[3][1]=pri[i]/100%10;
g[4][1]=pri[i]/10%10;
g[5][1]=pri[i]%10;
DFS(2);//搜索
}
if(pri[i]/10000>A)break;
}
sort(&ans[1],&ans[sum+1]);
for(int k=1;k<=sum;++k)
{
for(int i=1;i<=5;++i)
{
for(int j=1;j<=5;++j)
printf("%d",ans[k].g[i][j]);
printf("
");
}
printf("
");
}
if(sum==0)
cout<<"NONE"<<endl;
return 0;
}
然后看我的第二种解法：
首先还是找出所有满足条件的质数，
暴力求出他们的前缀和后缀（其实建立字典树也是可以的）
接下来，每次搜索一个格子（从第一行开始，自左向右，自上而下），
枚举填的数，并且判断是否可行（判断每一行是否存在前缀，每一列是否存在前缀，主对角线是否存在前缀，副对角线是否存在后缀），这样就能够减去大量的不合理情况。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<vector>
#include<algorithm>
using namespace std;
inline bool zs(int x)
{
int t=sqrt(x);
for(int i=2;i<=t;++i)
if(x%i==0)return false;
return true;
}
int S,A;
int g[6][6];
int hz[1000001],qz[1000001];
int pp[6]={1,10,100,1000,10000,100000};
int sum=0;
int L[10],R[10];
int djx1,djx2;
inline void outp()
{
for(int i=1;i<=5;++i)
{
printf("%d
",L[i]);
}
printf("
");
}
inline void Add(int x)//判断是否可行
{
int a[6]={0,x/10000,x/1000%10,x/100%10,x/10%10,x%10};
if(a[1]+a[2]+a[3]+a[4]+a[5]!=S)return;
if(!zs(x))return;
int i=x;
qz[i/10000]=qz[i/1000]=qz[i/100]=qz[i/10]=qz[i]=true;//前缀
hz[i%10000]=hz[i%1000]=hz[i%100]=hz[i%10]=hz[i]=true;//后缀
}
void DFS(int x,int y)
{
if(y==6)//换行
{
y=1;x+=1;
}
if(x==6)//输出解
{
++sum;
outp();
return;
}
bool ok1,ok2,ok3,ok4;
for(int i=0;i<=9;++i)//枚举当前数字
{
ok1=qz[L[x]*10+i];//行的值要在前缀中
ok2=qz[R[y]*10+i];//列的值要在前缀中
ok3=((x!=y)||(x==y&&qz[djx1*10+i]));//对角线的值要在前缀中
ok4=((x+y!=6)||(x+y==6&&hz[djx2+i*pp[x-1]]));//对角线的值要在后缀中
if(!(ok1&&ok2&&ok3&&ok4))continue;
L[x]=L[x]*10+i;
R[y]=R[y]*10+i;
if(x==y) djx1=djx1*10+i;
if(x+y==6)djx2=djx2+i*pp[x-1];
DFS(x,y+1); //向下搜索
L[x]/=10;
R[y]/=10;
if(x==y) djx1/=10;
if(x+y==6)djx2-=i*pp[x-1];
}
}
int main()
{
freopen("prime.in","r",stdin);
freopen("prime.out","w",stdout);
cin>>S;
cin>>A;
for(int i=10003;i<=99999;i+=2)
Add(i);
memset(g,-1,sizeof(g));
g[1][1]=A;
L[1]=R[1]=djx1=A;
DFS(1,2);
if(sum==0)
cout<<"NONE"<<endl;
return 0;
}
这种方法能够AC，但是效率很低。
AC归为AC
其实这题如果更改搜索顺序和得到更好地优化。
并且很明显的一点就是，如果求出了某 行/列/对角线 上的四个数，最后一个是可以直接计算判断的（效率更高）。【但是我很懒，没有继续优化了】