全排列及八皇后问题优化

之前一直困扰的全排列问题及八皇后问题，现在对这些问题的解法做一个记录。

全排列问题

#include<cstdio> 
#include<iostream>
using namespace std;
int vis[100] = { 0 };
int a[100];
int n;

void generate(int index) {
	if (index == n + 1) {
		for (int i = 1; i <= n; i++)
			cout << a[i];
		cout << endl;
	}
	for (int i = 1; i <= n; i++) {
		if (vis[i] == 0) {
			a[index] = i;
			vis[i] = 1;
			generate(index + 1);
			vis[i] = 0;
		}
	}
}
int main(void) {
	cin >> n;
	generate(1);
	return 0;
}

八皇后问题

八皇后问题就是把按照顺序记录每一列中皇后所在的行，及数组a的下标代表所在列的位置，a的值代表所在的行的位置，所以行列一定不会冲突，若两个的位置的下标之差的绝对值等于对应数组值之差的绝对值表示他们在一条对角线上。

#include<cstdio>
#include<iostream>
#include<math.h>
using namespace std;
int n;
int a[100];
int vis[100];
int count = 0;
void generate(int index){
	if(index == n + 1){
		for(int i = 1;i <= n;i++){
			for(int j = i + 1;j <= n;j++){
				if(abs(i - j) == abs(a[i] - a[j])){
					return;
				}
			}
		}
		count++;
	}
	for(int i = 1;i <= n;i++){
		if(vis[i] == 0){
			a[index] = i;
			vis[i] = 1;
			generate(index + 1);
			vis[i] = 0;
		}
	}
}
int main(void){
	cin >> n;
	generate(1);
	cout << count;
	return 0;
}

八皇后问题优化(回溯法)

下面是对上一种方法的优化，在上一种方法中，每次都是列出所有的位置后，判断是否有两个皇后在一条对角线上，下面我们每排列一个皇后就判断是否与之前的皇后在一条对角线上。降低了复杂度。

#include<cstdio>
#include<iostream>
#include<math.h>
using namespace std;
int n;
int a[100];
int vis[100];
int count = 0;
void generate(int index){
	if(index == n + 1){
		count++;
		return;
	}
	for(int i = 1;i <= n;i++){
		int flag = 1;
		if(vis[i] == 0){
			for(int j  = 1;j < index;j++){
				if(abs(j - index) == abs(a[j] - i)){
					flag = 0;
					break;	
				}
			}
			if(flag){
				a[index] = i;
				vis[i] = 1;
				generate(index + 1);
				vis[i] = 0;	
			}
		}
	}
}
int main(void){
	cin >> n;
	generate(1);
	cout << count;
	return 0;
}