八皇后问题
题目描述
一个如下的 6×6 的跳棋棋盘,有六个棋子被放置在棋盘上,使得每行、每列有且只有一个,每条对角线(包括两条主对角线的所有平行线)上至多有一个棋子。
上面的布局可以用序列 2 4 6 1 3 5 来描述,第 iii 个数字表示在第 iii 行的相应位置有一个棋子,如下:
行号 1 2 3 4 5 6
列号2 4 6 1 3 5
这只是棋子放置的一个解。请编一个程序找出所有棋子放置的解。
并把它们以上面的序列方法输出,解按字典顺序排列。
输入格式
一行一个正整数 n,表示棋盘是 n×n大小的。
输出格式
前三行为前三个解,每个解的两个数字之间用一个空格隔开。
第四行只有一个数字,表示解的总数。
输入输出样例
题解一:循环搜索
用一个数组x[i]表示第i层的皇后存储在第x[i]列,对于每一层,从1-n进行遍历,判断当前第a层的安排是否和前面1……a-1层的冲突,即是否在同一列或在对角线,如果冲突,就重新安排第a层,如果不冲突,进行递归安排第a+1层。结束条件就是所有层全部被安排完,即a>n时结束安排,输出结果
#include<bits/stdc++.h>
using namespace std;
int n,b=1;
int ans = 0;//方案数
int x[15];//表示第i层存储在第x[i]列
bool place(int b)
{
for (int i = 1; i <b; ++i)
{//如果第b层皇后和前面1……b-1层皇后都不在同一列且不在对角线的位置,就返回true
if(x[b]==x[i]||abs(b-i)==abs(x[b]-x[i])) return false;
}
return true;
}
void solve(int a)
{
if (a > n)
{
ans++;//方案数
if(b <= 3)//输出前三种放置方法
{
for (int i = 1; i < n; ++i)//为了最后一个元素不多输出空格,将x[n]单独输出
{
cout << x[i] << " ";
}
cout<<x[n];
cout << endl;
++b;
}
return ;
}
for (int i = 1; i <= n; ++i)
{
x[a] = i;//放置第a层皇后
if (place(a)) solve(a + 1);
}
}
int main()
{
cin >> n;
solve(1);
cout << ans;
return 0;
}
题解二:回溯
#include <bits/stdc++.h>
using namespace std;
int n;
int a[30], b[30], c[30],d[30];//行,列,斜上对角线,斜下对角线
int ans = 0,out=1;
void search(int t)
{
if (t > n)
{
ans++;
if (out <= 3)
{
for (int i = 2; i <= n; ++i)
cout << a[i] << " ";
cout << a[n+1] << endl;
out++;
}
return;
}
for (int j = 1; j <= n; ++j)
{//如果第j列并且两个对角线都无元素第j列就可以放
if ((!b[j]) && (!c[t + j]) && (!d[t - j + n]))
{
a[t + 1] = j;//第一个数在a[2]位置
b[j] = 1;
c[t + j] = 1;
d[t - j + n] = 1;
search(t + 1);
b[j] = 0;
c[t + j] = 0;
d[t - j + n] = 0;
}
}
}
int main()
{
cin >> n;
search(1);
cout << ans;
return 0;
}