试题 基础练习 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
题解:回溯法、C++
深度优先搜索(DFS,Depth-First Search) :是搜索的手段之一。它从某个状态开始,不断地转移状态直到无法转移,然后回退到前一步的状态,继续转移到其他状态,如此不断重复,直至找到
最终的解。例如求解数独,首先在某个格子内填入适当的数字,然后再继续在下一个格子内填入数字,如此继续下去。如果发现某个格子无解了,就放弃前一个格子上选择的数字,改用其他可行的数字。根据深度优先搜索的特点,采用递归函数实现比较简单。
回朔法 :是一种选优搜索法(试探法),被称为通用的解题方法,这种方法适用于解一些组合数相当大的问题。通过剪枝(约束+限界)可以大幅减少解决问题的计算量(搜索量)。
#include<iostream>
#include<cstdio>
using namespace std;
const int MAX_N = 8;
int n;
int ans = 0;
int c = 0;
int a[MAX_N][MAX_N];
int res1[3][MAX_N*2];//记录列,左对角线,右对角线是否有相同的皇后
int res2[3][MAX_N*2];//记录列,左对角线,右对角线是否有相同的皇后
void solve2(int cur) {
if(cur == n) { c++; return; }
for(int j = 0; j < n; j++) {
if(a[cur][j] && !res2[0][j] && !res2[1][cur+j] && !res2[2][n+cur-j]) {
res2[0][j] = 1; res2[1][cur+j] = 1; res2[2][n+cur-j] = 1;
solve2(cur+1);
res2[0][j] = 0; res2[1][cur+j] = 0; res2[2][n+cur-j] = 0;
}
}
return;
}
void solve1(int cur) {
if(cur == n) {
c = 0;
solve2(0);
ans += c;
return;
}
for(int j = 0; j < n; j++) {
if(a[cur][j] && !res1[0][j] && !res1[1][cur+j] && !res1[2][n+cur-j]) {
a[cur][j] = 0;
res1[0][j] = 1; res1[1][cur+j] = 1; res1[2][n+cur-j] = 1;
solve1(cur+1);
a[cur][j] = 1;
res1[0][j] = 0; res1[1][cur+j] = 0; res1[2][n+cur-j] = 0;
}
}
return;
}
int main() {
while(scanf("%d", &n) == 1) {
ans = 0;
for(int i = 0; i < n; i++)
for(int j = 0; j < n; j++)
scanf("%d", &a[i][j]);
solve1(0);
printf("%d\n", ans);
}
return 0;
}
回溯和深度优先搜索的区别
回溯是一种更通用的算法。可以用于任何类型的结构,其中可以消除域的部分 ——无论它是否是逻辑树。
深度优先搜索是与搜索树或图结构相关的特定回溯形式。它使用回溯作为其使用树的方法的一部分,但仅限于树/图结构。
回溯和 DFS 之间的区别在于回溯处理隐式树而 DFS 处理显式树。这似乎微不足道,但它意味着很多。当通过回溯访问问题的搜索空间时,隐式树在其中间被遍历和修剪。然而对于 DFS 来说,它处理的树/图是明确构造的,并且在完成任何搜索之前已经抛出了不可接受的情况,即修剪掉了。
因此,回溯是隐式树的 DFS,而 DFS 是回溯而不修剪。
————————————————————————————————————————————————————————
本题还用到了C语言连续输入问题,可参考我写的c语言连续输入篇
————————————————————————————————————————————————————————
题解二:C语言
#include<stdio.h>
int n;
int ans = 0;
int c = 0;
int a[8][8];
int res1[3][16];//记录列,左对角线,右对角线是否有相同的皇后
int res2[3][16];//记录列,左对角线,右对角线是否有相同的皇后
void solve2(int cur) {
int j;
if(cur == n) { c++; return; }
for(j = 0; j < n; j++) {
if(a[cur][j] && !res2[0][j] && !res2[1][cur+j] && !res2[2][n+cur-j]) {
res2[0][j] = 1; res2[1][cur+j] = 1; res2[2][n+cur-j] = 1;
solve2(cur+1);
res2[0][j] = 0; res2[1][cur+j] = 0; res2[2][n+cur-j] = 0;
}
}
return;
}
void solve1(int cur) {
if(cur == n) {
c = 0;
solve2(0);
ans += c;
return;
}
int j;
for(j = 0; j < n; j++) {
if(a[cur][j] && !res1[0][j] && !res1[1][cur+j] && !res1[2][n+cur-j]) {
a[cur][j] = 0;
res1[0][j] = 1; res1[1][cur+j] = 1; res1[2][n+cur-j] = 1;
solve1(cur+1);
a[cur][j] = 1;
res1[0][j] = 0; res1[1][cur+j] = 0; res1[2][n+cur-j] = 0;
}
}
return;
}
int main() {
while(scanf("%d", &n) == 1) {
ans = 0;
int i, j;
for(i = 0; i < n; i++)
for(j = 0; j < n; j++)
scanf("%d", &a[i][j]);
solve1(0);
printf("%d\n", ans);
}
return 0;
}