n皇后問題純數組解法

n皇后問題是一個典型的利用回溯法解決的問題,可以通過構建棧的方式來進行解決,但在這我將利用純數組的方式來實現n皇后問題的解法,大致思想是一致的

該代碼適合並未接觸數據結構的大一學生或自學學生來解決n皇后問題。

#include<iostream>
using namespace std;
int main()
{
	int num;
	cin >> num;
	int *a = new int[num];
	int i = 1, k, f, go = 1, count = 0;
	a[1] = 1;
	while (go)
	{
		while (go&&i <= num)
		{
			for (f = 1, k = 1; f&&k < i; k++)
				if (a[i] == a[k])
					f = 0;
			for (k = 1; f&&k < i; k++)
				if ((a[i] == a[k] - (k - i)) || (a[i] == a[k] + (k - i)))
					f = 0;
			if (!f)
			{
				if (a[i] == a[i - 1])
				{
					i--;
					if (i > 1 && a[i] == num)
						a[i] = 1;
					else if (i == 1 && a[i] == num)
						go = 0;
					else
						a[i]++;
				}
				else if (a[i] == num)
					a[i] = 1;
				else a[i]++;
			}
			else if (++i <= num)
				if (a[i - 1] == num)
					a[i] = 1;
				else a[i] = a[i - 1] + 1;
		}
		if (go)
		{
			++count;
			cout << "第" << count << "組:";
			for (k = 1; k <= num; k++)
				cout << a[k];
			cout << endl;
			if (a[num - 1] < num)
				a[num - 1]++;
			else
				a[num - 1] = 1;
			i = num - 1;
		}
	}
	delete[]a;
	return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章