八皇后问题的改进N皇后问题

N皇后问题

问题简述

经典的八皇后问题是在8×8格的国际象棋上摆放八个皇后，使其不能互相攻击，即任意两个皇后都不能处于同一行、同一列或同一斜线上，问有多少种摆法。现对该问题进行改良要求在一个n×n的棋盘上摆放n个皇后问有多少解法。

输入一个n，要求程序可以输出每一种解的摆放，并输出总的解法数量。

简单分析：

在n×n的棋盘上摆放n个皇后不能互相攻击，我们要做的就是找到这n个皇后各自的位置所以可以用一个数组来记录。又因为两个皇后不能放在一行所以每一行只能有一个皇后，问题就变成了需要确定每一行皇后的列数，我们可以用枚举法来递归回溯所有的可能。

C++代码：

#include<iostream>
#include<vector> 
#include<cmath>
using namespace std;
vector<int>QueenPos;  //定义一个int类型的容器用来存储皇后的位置 
int n,counts=0;		  //counts记录方法数量 
void Nqueen(int k)    //尝试第k行每列的可能性 
{
	if(k==n)//因为 QueenPos下标从0开始所以k=n时说明n个皇后已全部摆放 
	{
		counts++;    //解法加一 
		printf("第%d种方案:\n",counts);
		for(int i = 0; i < n; i++)   //输出这种方案
		{
			for(int j = 0; j < n; j++)
			{
				if(j==QueenPos[i])	//有皇后的位置输出X
					cout<<"X ";
				else				//没有皇后的位置输出o  
					cout<<"o ";
			}
			cout<<endl;
		}
		cout<<endl;
	}
	else
	{								//若k小于n说明还需摆放k行的皇后 
		for(int i = 0; i < n; i++) //尝试每一列的可能性 
		{
			int j;
			for(j = 0; j < k; j++) //判断此列会不会和前面k-1个皇后冲突 
			{	//即判断和之前的皇后在不在同一列 或 一条斜线(即行列差绝对值相等) 
				if((QueenPos[j]==i)||(abs(QueenPos[j]-i)==abs(j-k)))
					break;
			}
			if(j==k)
			{					//j=k说明第i列和前面的位置不冲突
				QueenPos[k]=i;  //便将其记录  
				Nqueen(k+1);    //继续摆放下一行 
			}
		}
	}
}
int main()
{
	cin>>n;					//输入规模 
	QueenPos.resize(n);		//设置一下容器长度 
	Nqueen(0);				//从第0行开始摆放 
	printf("总计有%d种方案!",counts); //打印总解决数 
	return 0;
}

相同思想的Python代码：

QueenPos = list()   #定义一个列表用来存储皇后的位置
n = counts = 0      #counts记录方法数量
def dfs(k):         #尝试第k行每列的可能性
    if k==n:        #终止条件
        global counts #声明全局变量进行解法累加
        counts+=1
        print("第%d种方案:" % counts)
        for i in range(n):#输出方案
            for j in range(n):
                if j==QueenPos[i]:
                    print("X ",end="")
                else:
                    print("o ", end="")
            print()
        print()
    else:              #尝试判断并递归
        for i in range(n):
            t = 0
            for j in range(k):
                if(QueenPos[j]==i) or (abs(QueenPos[j]-i)==abs(j-k)):
                    break;
                t+=1
            if t == k:
                QueenPos[k] = i;
                dfs(k + 1);
n = int(input())        #输入规模
for i in range(n):
    QueenPos.append(0)
dfs(0)                  #开始摆放
print("总计有%d种方案!"%counts)

测试执行：