相信大家对N皇后问题一定不陌生,网上的代码都是基于斜率来做的,也就是|y1-y2|=|x1-x2|,根据数学知识,理解这个做法并不难,今天老师讲课时,提到了另一种写法,便回来尝试了一下,发现写法比上面那种复杂了一点,但是也算是一个收获吧,自己动手写印象更深刻~
我们知道|y1-y2|=|x1-x2|,其实也就是x1+y1=x2+y2或者x1-y1=x2-y2,不妨就定义这么两个数组,如果新来的x,y相加或者相减等于这两个数组中的数,说明不可以放进去,反之则可以
#include <iostream>
using namespace std;
int n;
int a[101];
int col[101],pie[101],na[101];
bool judge(int N) {
for (int j=1;j<N;j++) {
if (col[j]==col[N] || pie[j]==pie[N] || na[j]==na[N]) {
return false;
}
}
return true;
}
void fun(int i) {
if (i>n) {
for (int j=1;j<=n;j++) {
cout << j << " " << a[j] << endl;
}
cout << endl;
}
else {
for (int j=1;j<=n;j++) {
a[i]=j;
col[i]=j;
pie[i]=i+j;
na[i]=i-j;
if(judge(i)) {
fun(i+1);
}
}
}
}
int main() {
cin >> n;
fun(1);
return 0;
}