打表法详解以及n皇后详解
描述
在N*N的方格棋盘放置了N个皇后,使得它们不相互攻击(即任意2个皇后不允许处在同一排,同一列,也不允许处在与棋盘边框成45角的斜线上。
你的任务是,对于给定的N,求出有多少种合法的放置方法。
输入
1
8
5
0
输出
1
92
10
详解
经典的n皇后问题,使用回溯法求解,可是非常容易超时,所以用打表进行优化
n皇后求解呢,总结起来思路是这样:
其实用不着二维数组,因为每一行只能放一个皇后,不然就不符合条件了是不是,所以我们只要开一个一维数组,下表是皇后所在行,值是皇后所在列就好了,二维降到一维运算量能减少很多呢,然后我们进行搜索回溯,现在第一行第一列放置,然后再去第二行第一列开始到第n列,找到能放置皇后的位置进行放置,直到n行,如果放了n个皇后,那么计数cot+1,如果不到则回去上一个状态然后再进行搜索。因为本质上其实也是一种暴力搜索发,所以要进行打表优化时间。
浅层的打表意思是:因为每次输入,它都要进行运算一次,可能会有重复的,那么我们事先提前算好,然后在输出。
代码
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
using namespace std;
int n,queen[20],num;
bool place(int key)
{
for(int i=1; i < key; i++)
if(abs(key - i) == abs(queen[i] - queen[key]) ||
queen[key] == queen[i])
return false;
return true;
}
void backtrack(int x)
{
if(x>n)
num++;
else
for(int i=1;i<=n;i++)
{
queen[x]=i;
if(place(x))
backtrack(x+1);
}
}
int ans[11];
int main()
{
//事先打表
for(int i=1;i<=10;i++)
{
num=0;
n=i;
memset(queen,0,20);
backtrack(1);
ans[i]=num;
}
while(1){
cin>>n;
if(n==0) break;
cout<<ans[n];
}
return 0;
}
但是因为是这题的数据量比较小,如果遇到了n再大一点的情况,即使这种打表也会TLE,那怎么办呢
比如
int main()
{
clock_t s,e;
s=clock();
for(int i=1;i<=14;i++)
{
num=0;
n=i;
memset(queen,0,20);
backtrack(1);
printf("%d,",num);
}
e=clock();
printf("\nRuntime:%.3fs\n", ((double) (e - s))/CLK_TCK);
return 0;
}
1,0,0,2,10,4,40,92,352,724,2680,14200,73712,365596,
Runtime:11.143s
当n到14的时候,光是这种打表就要11.143s,那么我们怎么办呢
我们这时候要用到c++的一个简便的文件读写
freopen("out.txt","w",stdout);
这个可以替代标准输出,直接写入文件,那我们怎么用呢?
//上面的函数一样的,不在展示
int main()
{
freopen("out.txt","w",stdout);
clock_t s,e;
s=clock();
for(int i=1;i<=15;i++)
{
num=0;
n=i;
memset(queen,0,20);
backtrack(1);
printf("%d,",num);
}
e=clock();
printf("\nRuntime:%.3fs\n", ((double) (e - s))/CLK_TCK);
return 0;
}
我们把数据扩展到15,然后对文件输出。点击运行,等待程序结束。这时候只要不是数据或者内存爆掉,终究是能够算出来的。
程序结束后我们到源代码的根目录,找到out.txt
1,0,0,2,10,4,40,92,352,724,2680,14200,73712,365596,2279184,
Runtime:76.174s
发现14到15只多了1但是时间多了7倍还多呢,这时候我们可以复制前面的数值,记得去除末尾逗号和前面添加0
int ans[16]={0,1,0,0,2,10,4,40,92,352,724,2680,14200,73712,365596,2279184};
int main()
{
while(1){
cin>>n;
if(n==0)break;
cout<<ans[n]<<endl;
}
return 0;
}
那么我们打表就完成了,运行也不会超时了。