参考书籍《算法竞赛入门经典》刘汝佳
设置一个数组int c[maxn] ,c[i] = k 表示第i行第k列放了一个皇后,因此,原问题可以表示为对0-n进行全排列(对于八皇后问题,n是7),在这些排列中筛选出符合条件,即互不攻击的排列放入数组c
那么怎么判断c[cur]是否与当前已经放入的皇后发生冲突呢,这里设置了一个visit[3][maxn*2]的数组,visit[0][i] = 1表示第i列有皇后了,visit[1][i]表示i对应的主对角线有皇后了,visit[2][i]表示i对应的副对角线有皇后了,我们可以用x-y+n来代表一条主对角线,x+y代表副对角线,对于4*4的方格,主对角线标志如图1,副对角线标志如图2,因此,如果!visit[0][i] && !visit[1][cur+i] && !visit[2][cur-i+n]返回1,则说明(cur,i)点所在的行,列,主对角线,副对角线都没有皇后存在,则可以在这里放皇后,然后递归下一行
4 | 3 | 2 | 1 |
5 | 4 | 3 | 2 |
6 | 5 | 4 | 3 |
7 | 6 | 5 | 4 |
图1
0 | 1 | 2 | 3 |
1 | 2 | 3 | 4 |
2 | 3 | 4 | 5 |
3 | 4 | 5 | 6 |
#include<iostream>
using namespace std;
const int maxn = 100; //这个随便多大,反正100皇后问题也不过分
int c[maxn];
int n;
int visit[3][2*maxn];
void print()
{
for (int i = 0; i < n; i++)
{
cout<<"("<<i<<", "<<c[i]<<") ";
}
cout<<endl;
}
void queen_search(int cur)
{
if(cur == n) print();
else for(int i = 0; i < n; i++)
{
if(!visit[0][i] && !visit[1][cur+i] && !visit[2][cur-i+n]) //x-y+n可以唯一标识出主对角线,x+y可以唯一标识出副对角线
{
c[cur] = i;
visit[0][i] = visit[1][cur+i] = visit[2][cur-i+n] = 1;
queen_search(cur + 1);
visit[0][i] = visit[1][cur+i] = visit[2][cur-i+n] = 0;
}
}
}
int main()
{
while(cin >> n)
{
memset(visit, 0, sizeof(visit) );
queen_search(0);
}
}