题目
试题 基础练习 2n皇后问题
资源限制
时间限制:1.0s 内存限制:512.0MB
问题描述
给定一个n*n的棋盘,棋盘中有一些位置不能放皇后。现在要向棋盘中放入n个黑皇后和n个白皇后,使任意的两个黑皇后都不在同一行、同一列或同一条对角线上,任意的两个白皇后都不在同一行、同一列或同一条对角线上。问总共有多少种放法?n小于等于8。
输入格式
输入的第一行为一个整数n,表示棋盘的大小。
接下来n行,每行n个0或1的整数,如果一个整数为1,表示对应的位置可以放皇后,如果一个整数为0,表示对应的位置不可以放皇后。
输出格式
输出一个整数,表示总共有多少种放法。
样例输入
4
1 1 1 1
1 1 1 1
1 1 1 1
1 1 1 1
样例输出
2
样例输入
4
1 0 1 1
1 1 1 1
1 1 1 1
1 1 1 1
样例输出
0
先放总代码,在分段慢慢讲,毕竟这个题我写了3个小时,终于通过了!!!好激动,我要好好记下来,方便自己复习,也方便大家看懂,总代码:
#include
using namespace std;
int queen[8][8];
int n;
int sum=0;
void queen2(int line);//白子
bool check(int line,int list,int a);//检查此处是否可以放置棋子
void queen1(int line);//黑子
int main()
{
cin >> n;
for(int i = 0;i < n;i ++)
{
for(int j = 0;j < n;j ++)
{
cin >> queen[i][j];
}
}
queen1(0);
cout << sum;
return 0;
}
//0--不可以放置,1--是可以放置,2--是放置黑子 ,3--是放置白子
bool check(int line,int list,int a)
{
int i,j;
if(queen[line][list] != 1) return false;//要检查的位子为line行list列,若此处初始值不为1则不可以放置棋子
else//line行list列处初始值为1
{
for(i = 0;i < line;i ++)//遍历line行之前的行数
{
if(queen[i][list] != a)//若list列上没有相同的子
{
for(j = 0;j < n;j ++)//继续遍历i行
{
if(queen[i][j] == a )//如果i行j列上有相同的子 (a=2为黑子,a=3为白子)
if( line + list == i + j || line - list == i - j)//并且i行j列上面的子与要判断的位置line行list列在对角线上
return false;//返回false,即line行list列不可放置
}
}
else if(queen[i][list] == a) return false;//list列上有相同的子,返回false
}
}
return true;//检查无误,返回true
}
void queen1(int line)
{
int j;
for(j = 0;j < n;j ++)
{
if(queen[line][j] == 1 && check(line,j,2))//如果i行j列为1,并且检查无互相攻击表示可以放置
{
queen[line][j] = 2;//则放置黑子,数字为2
if(line == n-1)//黑子在棋盘上放完
{
queen2(0);//开始放白子
queen[line][j] = 1;//回溯之后值置1
return ;
}
queen1(line + 1);
queen[line][j] = 1;//回溯之后值置1
}
}
return ;
}
void queen2(int line)
{
int j;
for(j = 0;j < n;j ++)
{
if(queen[line][j] == 1 && check(line,j,3))//如果i行j列为1,并且检查无互相攻击表示可以放置
{
queen[line][j] = 3;//则放置白子,数字为3
if(line == n - 1)
{
sum ++;
queen[line][j] = 1;
return ;
}
queen2(line + 1);
queen[line][j] = 1;//还原
}
}
return ;
}
头文件和声明部分:
#include
using namespace std;
int queen[8][8];
int n;
int sum=0;
void queen2(int line);//白子
bool check(int line,int list,int a);//检查此处是否可以放置棋子
void queen1(int line);//黑子
queen[8][8]:因为题目中说n小于等于8,所以最大就是8了,就定义一个8*8的数组。
int n:由用户自己输入n,最后是n*n的棋盘,因为好几个函数都要用到,放在全局变量就不用形参实参传来传去了,方便。
int sum = 0:用来计算满足条件的放置棋子的方式的总数。
queen1:此函数用来放置黑子
queen2:此函数用来放置白子
check:函数用来检查line行list列是否可以放置棋子
接下来讲main函数:
int main()
{
cin >> n;
for(int i = 0;i < n;i ++)
{
for(int j = 0;j < n;j ++)
{
cin >> queen[i][j];
}
}
queen1(0);
cout << sum;
return 0;
}
for循环:用来输入盘的,1–可以放置棋子,0–不可以放置棋子
queen1(0):从第0行开始判断每个列哪里放置黑子(白子的入口不在main函数,要黑子都放完了才开始重新遍历放置白子,所以白的入口在queen1里面,之后讲)
check函数:
//0--不可以放置,1--是可以放置,2--是放置黑子 ,3--是放置白子
bool check(int line,int list,int a)
{
int i,j;
if(queen[line][list] != 1) return false;//要检查的位子为line行list列,若此处初始值不为1则不可以放置棋子
else//line行list列处初始值为1
{
for(i = 0;i < line;i ++)//遍历line行之前的行数
{
if(queen[i][list] != a)//若list列上没有相同的子
{
for(j = 0;j < n;j ++)//继续遍历i行
{
if(queen[i][j] == a )//如果i行j列上有相同的子 (a=2为黑子,a=3为白子)
if( line + list == i + j || line - list == i - j)//并且i行j列上面的子与要判断的位置line行list列在对角线上
return false;//返回false,即line行list列不可放置
}
}
else if(queen[i][list] == a) return false;//list列上有相同的子,返回false
}
}
return true;//检查无误,返回true
}
queen[line][list]的值有4种:0–不可以放置棋子,1–可以放置任意棋子(只是说可以放,但是具体放不放还要看相同的子(皇后)能不能互相攻击),2–放置了黑子,3–放置了白子。
形参a就是用来判断使黑子还是白子的,如果是黑子来判断,a=2,如果是白子来判断a=3
其他注释上有,不说了