深度优先搜索
深度优先搜索算法(英语:Depth-First-Search,简称DFS)是一种用于遍历或搜索树或图的算法。沿着树的深度遍历树的节点,尽可能深的搜索树的分支。当节点v的所在边都己被探寻过,搜索将回溯到发现节点v的那条边的起始节点。这一过程一直进行到已发现从源节点可达的所有节点为止。如果还存在未被发现的节点,则选择其中一个作为源节点并重复以上过程,整个进程反复进行直到所有节点都被访问为止。属于盲目搜索。[wikipedia]
关键词:深度优先搜索-深搜-dfs-递归-回溯
此题目为 N皇后问题 变形。
代码如下:
#include <iostream>
#include <cstring>
#define MAXN 10
using namespace std;
int __MAP[MAXN][MAXN], line[MAXN],
tot, num, n, k;
void dfs(int i)
{
if (k == num) //搜到全部棋子都放到棋盘上
{
tot++;
return;
}
if (i == n) //搜到最后一行没有办法继续搜
return;
for (int j = 0; j < n; j++)
{
if (line[j] && __MAP[i][j])
{
line[j] = 0;
num++;
dfs(i + 1); //搜索下一行
line[j] = 1; //回溯
num--;
}
}
dfs(i + 1); //当前行不能放棋子时直接放下一行
}
int main()
{
int i, j;
char c;
while(cin >> n >> k)
{
if (-1 == n && -1 == k) break;
for (i = 0; i < n; i++)
{
for (j = 0; j < n; j++)
{
cin >> c;
__MAP[i][j] = ('#' == c) ? 1 : 0;
}
}
memset(line, 1, sizeof(line));
tot = num = 0;
dfs(0);
cout << tot << endl;
}
return 0;
}