搜索之DFS

最新推荐文章于 2024-10-05 12:07:50 发布

sss-yyyy

最新推荐文章于 2024-10-05 12:07:50 发布

阅读量85

点赞数 2

文章标签： c++ 算法

本文链接：https://blog.csdn.net/yituanzao___/article/details/125964577

版权

一、DFS深度优先搜索（递归）

深搜说白了就是一条路走到黑。先访问初始顶点，若当前访问的点有还未被访问的邻接顶点，则任选一个访问；反之，则回溯到最近一次访问过的顶点，直至所有顶点都被访问过一次。

例1.全排列

题目描述

按照字典序输出自然数 1 到 n 所有不重复的排列，即 n 的全排列，要求所产生的任一数字序列中不允许出现重复的数字。

输入格式

一个整数 n。

输出格式

由 1∼n 组成的所有不重复的数字序列，每行一个序列。

每个数字保留 5 个场宽。

输入输出样例

输入 #1

输出 #1

    1    2    3
    1    3    2
    2    1    3
    2    3    1
    3    1    2
    3    2    1

说明/提示

1≤n≤9。

思路：把题目理解成是选数字，每回选的时候都在未选过的里面选，选完之后用数组标记，完成一种排列后，再回溯，将标记选过的数组状态改变，重新选择，递归下去......

代码：

#include<bits/stdc++.h>
using namespace std;

int n, ans[12];//ans数组用来存答案
bool step[12];//保存状态（即该数是否已经被选过）

void dfs(int u) {
	if (u == n + 1) {//如果u > n，则已经完成一次排列
		for (int i = 1; i <= n; i++) printf("%5d", ans[i]);
		cout << endl;
		return;
	}

	for (int i = 1; i <= n; i++) {
		if (!step[i]) {//如果这个数还未被选过
			ans[u] = i;
			step[i] = true;//标记为已经选过
			dfs(u + 1);递归
			step[i] = false;//改变状态回溯
		}
	}
}

int main() {
	cin >> n;
	dfs(1);
	return 0;
}

例2：选数

题目描述

已知 n 个整数 x1,x2,⋯,xn，以及 1 个整数 k（k<n）。从 n 个整数中任选 k 个整数相加，可分别得到一系列的和。例如当 n=4，k=3，4 个整数分别为3,7,12,19 时，可得全部的组合与它们的和为：

3+7+12=22

3+7+19=29

7+12+19=38

3+12+19=34

现在，要求你计算出和为素数共有多少种。

例如上例，只有一种的和为素数：3+7+19=29。

输入格式

第一行两个空格隔开的整数 n,k（1≤n≤20，k<n）。

第二行 n 个整数，分别为 x1,x2,⋯,xn（1≤xi≤5×10^6）。

输出格式

输出一个整数，表示种类数。

输入输出样例

输入 #1

4 3
3 7 12 19

输出 #1

思路：与上一题大致相同，不过就是多加一个判断素数的函数，要注意的是上一题是要排列，这一题是组合数，最后要记得将答案除以k的阶乘

代码：

#include<bits/stdc++.h>
using namespace std;

int n, k, a[25], b[25];
bool step[25];

bool isPrime(int x) {//判断和是否为素数
	if (x == 1 || x == 2) return true;
	for (int i = 2; i < x; i++) {
		if (x % i == 0) return false;
	}
	return true;
}

int sum = 0, ans = 0;
void dfs(int cnt) {//求出所有组合的和
	if (cnt == k + 1) {
		sum = 0;
		for (int i = 1; i <= k; i++) {
			sum += b[i];
		}
		if (isPrime(sum)) ans += 1;//判断和是否为素数
		return;
	}

	for (int i = 1; i <= n; i++) {
		if (!step[i]) {
			b[cnt] = a[i];
			step[i] = true;
			dfs(cnt + 1);
			step[i] = false;
		}
	}
}

long long f(int y) {
	if (y == 1) return 1;
	return y * f(y - 1);
}

int main() {
	cin >> n >> k;
	for (int i = 1; i <= n; i++) cin >> a[i];
	dfs(1);
	ans /= f(k);//一定注意最后要除以k的阶乘
	cout << ans;
	return 0;
}

求连通块类型

例3：求细胞数量

题目描述

一矩形阵列由数字 0 到 9 组成，数字 1 到 9 代表细胞，细胞的定义为沿细胞数字上下左右若还是细胞数字则为同一细胞，求给定矩形阵列的细胞个数。

输入格式

第一行两个整数代表矩阵大小 n 和 m。

接下来 n 行，每行一个长度为 m 的只含字符 0 到 9 的字符串，代表这个 n×m 的矩阵。

输出格式

一行一个整数代表细胞个数。

输入输出样例

输入 #1

输出 #1

说明/提示

数据规模与约定

对于 100% 的数据，保证1≤n,m≤100。

思路：当读到不为0的数时，记录到答案中，然后DFS搜索到该块数据并全部记为0，防止重复读入

#include<bits/stdc++.h>
using namespace std;

int n, m;
int a[105][105];
int fx[] = { -1,0,1,0 }, fy[] = { 0,-1,0,1 };

void dfs(int x, int y) {
	a[x][y] = 0;
	for (int i = 0; i < 4; i++) {
		x += fx[i], y += fy[i];
		if (x > 0 && x <= n && y > 0 && y <= m && a[x][y]) {
			dfs(x, y);
		}
		x -= fx[i], y -= fy[i];
	}
}

int main() {
	int cnt = 0;
	cin >> n >> m;
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= m; j++) {
			scanf("%1d", &a[i][j]);
		}
	}
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= m; j++) {
			if (!a[i][j]) continue;
			dfs(i, j);//若不为0，则DFS搜周围所有不为0的数将其全部归0
			cnt++;
		}
	}
	cout << cnt;
	return 0;
}

总结：深搜比起宽搜较为节省空间，但是费时间（因为深搜是将每种答案都记录了下来），更适合找所有答案的题