在3*3个方格的方阵中填入数字1-10的某9个数字,每个方格填一个整数,使所有相邻两个方格内的两个整数之和为素数。编写一个程序,求出所有满足这个要求的数字的填法。
#include<iostream>
#include<cmath>
using namespace std;
int a[11]={0,1,2,3,4,5,6,7,8,9,10};
int x[11];
int sum=0;
bool unprime(int a,int b)
{
int c=a+b,counter=0;
for(int i=2;i<=sqrt(c);i++)
{
if(c%i==0)counter++;
}
if(counter!=0)return true;
else return false;
}
void func(int i)
{
if(i==11)
{
for(int i=1;i<9;i++)
{
if(unprime(x[i],x[i+1]))return;
}
if(unprime(x[8],x[1])||unprime(x[9],x[2])||unprime(x[9],x[4])||unprime(x[9],x[6]))return;
sum++;
for(int i=1;i<=3;i++)cout<<x[i]<<' ';
cout<<endl;
cout<<x[8]<<' '<<x[9]<<' '<<x[4]<<endl;
cout<<x[7]<<' '<<x[6]<<' '<<x[5]<<endl<<endl<<endl;
}
else
{
for(int j=i;j<=10;j++)
{
swap(a[i],a[j]);
x[i]=a[i];
func(i+1);
swap(a[i],a[j]);
}
}
}
int main()
{
func(1);
cout<<"总数:"<<sum;
return 0;
}
- 采用回溯法,即求全排列,时间复杂度O(n!).