DFS(深度优先搜索)
概念
数据结构:stack
空间:O(h) —— h为树的高度
不具备最短性
例题及代码
例题1:按字符序来打印数字1~n
用深度优先搜索从1开始一层一层搜索就能得到答案。
代码
#include <iostream>
using namespace std;
const int N = 10;
int n;
int path[N];
bool used[N];
void dfs(int u)
{
if (u == n) {
for (int i = 0; i < n; i++) cout << path[i] << ' ';
cout << endl;
return;
}
for (int i = 1; i <= n; i++) {
if (!used[i]) {
path[u] = i;
used[i] = true;
dfs(u + 1);
used[i] = false;
}
}
}
int main()
{
cin >> n;
dfs(0);
return 0;
}
例题2:n皇后问题(在n*n的棋盘中,打印出n个皇后能够同时存在的棋盘布置。皇后的攻击范围是同一行列,和两条对角线)
代码
#include <iostream>
using namespace std;
const int N = 20;
int n;
char g[N][N];
bool col[N], dg[N], udg[N];
void dfs(int u)
{
if (u == n) {
for (int i = 0; i < n; i++) cout << g[i] << endl;
cout << endl;
return;
}
for (int i = 0; i < n; i++) {
//判断是否在同一列、两条对角线当中
if (!col[i] && !dg[u + i] && !udg[n - u + i]) {
g[u][i] = 'Q';
col[i] = dg[u + i] = udg[n - u + i] = true;
dfs(u + 1);
col[i] = dg[u + i] = udg[n - u + i] = false;
g[u][i] = '.';
}
}
}
int main()
{
cin >> n;
for (int i = 0; i < n; i++)
for (int j = 0; j < n; j++)
g[i][j] = '.';
dfs(0);
return 0;
}
BFS(宽度优先搜索)
概念
数据结构:queue
空间:O(2^h)
具备最短性
例题及代码
例题:从0,0点出发到右下角最少需要几步
宽度优先搜索能够保证第一次搜到的这个点就是最短路
代码:
#include <cstring>
#include <iostream>
using namespace std;
typedef pair<int, int> PII;
const int N = 110;
int n, m;
int g[N][N];
int d[N][N];//表示每个点到起点的距离
PII q[N * N], pre[N][N];
int bfs()
{
int hh = 0, tt = 0;
q[0] = { 0, 0 };
//初始化距离为-1,表示没走过
memset(d, -1, sizeof d);
d[0][0] = 0;
int dx[4] = { -1, 0, 1, 0 };
int dy[4] = { 0, 1, 0, -1 };
while (hh <= tt) {
auto t = q[hh++];
for (int i = 0; i < 4; i++) {
int x = t.first + dx[i];
int y = t.second + dy[i];
if (x >= 0 && x < n && y >= 0 && y < m && g[x][y] == 0 && d[x][y] == -1) {
d[x][y] = d[t.first][t.second] + 1;
pre[x][y] = t;
q[++tt] = { x, y };
}
}
}
//查看路径
int x = n - 1, y = m - 1;
while (x || y) {
cout << x << ' ' << y << endl;
auto t = pre[x][y];
x = t.first, y = t.second;
}
return d[n - 1][m - 1];
}
int main()
{
cin >> n >> m;
for (int i = 0; i < n; i++)
for (int j = 0; j < m; j++)
cin >> g[i][j];
cout << bfs() << endl;
return 0;
}