N皇后问题
用深度优先搜索,直接套用模板就好。
注意mark函数和unmark函数
mark函数把格子中的值-1,其中unmark函数,把格子中的值+1,为什么不是直接改变值,是因为有可能多个皇后都攻击那一个位置,如果直接改变值,那么可能会影响某个皇后的攻击范围。
代码
#include<iostream>
#include<cstring>
using namespace std;
const int max_num=100;
int n;
int s[max_num][max_num];
struct queue{
int id;
int x;
int y;
}Q[max_num];
//将一个皇后的攻击范围标记出来,皇后的坐标为x,y
void mark(int x,int y)
{
//攻击同一行
for(int i=1;i<=n;i++)
{
s[x][i]--;
}
//攻击同一列
for(int i=1;i<=n;i++)
{
s[i][y]--;
}
//攻击右上
int xl=x+1;
int yl=y+1;
while(xl>=1&&xl<=n&&yl>=1&&yl<=n)
{
s[xl][yl]--;
xl++;
yl++;
}
//攻击右下
xl=x+1;
yl=y-1;
while(xl>=1&&xl<=n&&yl>=1&&yl<=n)
{
s[xl][yl]--;
xl++;
yl--;
}
//攻击左上
xl=x-1;
yl=y+1;
while(xl>=1&&xl<=n&&yl>=1&&yl<=n)
{
s[xl][yl]--;
xl--;
yl++;
}
//攻击左下
xl=x-1;
yl=y-1;
while(xl>=1&&xl<=n&&yl>=1&&yl<=n)
{
s[xl][yl]--;
xl--;
yl--;
}
s[x][y]=1;
}
//撤销皇后的攻击范围
void unmark(int x,int y)
{
//攻击同一行
for(int i=1;i<=n;i++)
{
s[x][i]++;
}
//攻击同一列
for(int i=1;i<=n;i++)
{
s[i][y]++;
}
//攻击右上
int xl=x+1;
int yl=y+1;
while(xl>=1&&xl<=n&&yl>=1&&yl<=n)
{
s[xl][yl]++;
xl++;
yl++;
}
//攻击右下
xl=x+1;
yl=y-1;
while(xl>=1&&xl<=n&&yl>=1&&yl<=n)
{
s[xl][yl]++;
xl++;
yl--;
}
//攻击左上
xl=x-1;
yl=y+1;
while(xl>=1&&xl<=n&&yl>=1&&yl<=n)
{
s[xl][yl]++;
xl--;
yl++;
}
//攻击左下
xl=x-1;
yl=y-1;
while(xl>=1&&xl<=n&&yl>=1&&yl<=n)
{
s[xl][yl]++;
xl--;
yl--;
}
s[x][y]=0;
}
void dfs(int q)
{
if(q==n+1)
{
for(int i=1;i<=n;i++)
{
cout<<"第"<<i<<"个皇后的位置为"<<endl;
cout<<"横坐标"<<Q[i].x<<' '<<"纵坐标"<<Q[i].y<<endl;
}
return ;
}
else{
for(int i=1;i<=n;i++)
{
if(s[q][i]==0)
{
mark(q,i);
Q[q].x=q;
Q[q].y=i;
dfs(q+1);
unmark(q,i);
Q[q].x=0;
Q[q].y=0;
}
}
}
return ;
}
int main()
{
cout<<"皇后的数量"<<endl;
cin>>n;
memset(s,0,sizeof(s));
dfs(1);
return 0;
}