这题写了几个版本,第一次写超时,于是用一个数组保存当前路径,这样对于判断是否可以放置有利,但是还是超时。
最后发现是之所以会超时,是因为测试的例子非常多,应该使用预处理,如果每次都重新计算会超时,只要把结果保存在数组,然后每次查询即可。
代码最终版本:
#include <iostream>
using namespace std;
const int N = 11;
int map[N][N];
int cnt;
int n;
int c[N];//保存当前解
int a[N];//保存预处理结果
bool place(int k)
{
for(int i = 0; i < k; i++)
{
if(abs(k - i) == abs(c[k] - c[i]) || c[i] == c[k])
return false;
}
return true;
}
void backtrack(int k)
{
if(k >= n)
{
cnt++;
// return;
}
else
{
for(int i = 0; i < n; i++)
{
c[k] = i;
if(place(k))
{
map[k][i] = 1;
backtrack(k+1);
map[k][i] = 0;
}
}
}
}
int main()
{
for(int i = 1; i < N; i++)
{
memset(map, 0, sizeof(map));
memset(c, 0, sizeof(c));
cnt = 0;
n = i;
backtrack(0);
a[i] = cnt;
}
while(cin >> n, n)
{
cout << a[n] << endl;
}
return 0;
}