DFS入门

最近有一场机试,已经说了重点考察dfs,但是对dfs还不是很熟,所以借由学习dfs来输出笔记,从而加深印象。

一.概念

dfs,深度搜索算法,又可以认为是回溯算法,它其实就是一个决策树的遍历问题,遍历出所有情况,但是先深度后广度。用另外一个名称说这个问题其实就是穷举,穷举能做出的所有选择。

那么其实就是去思考三个问题:

  1. 当前做的选择
  2. 手上可以做的选择
  3. 什么时候不能做选择了?就是达到最后一个叶子结点的时候

最经典的问题当属全排列问题,如何排列出1-n所有数字,按字典顺序。拿acwing y神的一张图就是这样子写草稿:

那么我们可以看出要分别去做不同的选择最终得到所有结果,光讲可能抽象了点,我们看道题目

排列数字

给定一个整数 n,将数字 1∼n 排成一排,将会有很多种排列方法。

现在,请你按照字典序将所有的排列方法输出。

输入格式

共一行,包含一个整数 n。

输出格式

按字典序输出所有排列方案,每个方案占一行。

数据范围

1≤n≤7

输入样例:

复制代码
3

输出样例:

复制代码
1 2 3
1 3 2
2 1 3
2 3 1
3 1 2
3 2 1

解析与代码

那么对于这道题结合上面那张图我们就知道首先要从1出发,然后去遍历除了1的1-n其他数字,那么1就要有个状态,记录已经遍历过该数字了,1就要添加到路径之中。

后面去遍历其他数字,其他数字如果也遍历了也就是说在路径里面那么其实就没有必要去遍历了。因为是要全搜索,所以最后我们也一定要把原先遍历的去除掉,把状态删掉

java 复制代码
import java.util.*;

public class Main {
	static int n;
	static int[] path;
	static boolean[] state;
	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		n = in.nextInt();
		path = new int[n + 1];
		state = new boolean[n + 1];
		dfs(1);
	}
	public static void dfs(int u) {
		if(u > n) {
			for(int i = 1;i <= n;i ++) {
				System.out.print(path[i] + " ");
			}
			System.out.println();
			return;
		}
		for(int i = 1;i <= n;i ++) {
			if(!state[i]) {
				path[u] = i;
				state[i] = true;
				dfs(u + 1);
				state[i] = false;
			}
		}
		
	}
}

总结出一个框架

java 复制代码
result = [];
void dfs(路径, 选择列表) {
    if(满足结束条件):
        result.addd(路径);
        return;
    
    for 选择 in 选择列表 {
        做选择
        dfs(路径,选择列表)
        去除此选择
    }
}

n-皇后

这个图解思路我觉得写的很好,解答了为什么dg和udg要那么设置函数

java 复制代码
import java.util.*;
public class Main {
	static int n;
	static char[][] chs; // 答案
	static boolean[] col, dg, udg;  // 状态
	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		n = in.nextInt();
		chs = new char[n + 1][n + 1];
		col = new boolean[n + 10];
		dg = new boolean[n + 10];
		udg = new boolean[n + 10];
		for(int i = 0;i < n;i ++) {
			for(int j = 0;j < n;j ++) {
				chs[i][j] = '.';
			}
		}
		dfs(0);
	}
	public static void dfs(int x) {
		if(x == n) {
			for(int i = 0;i < n;i ++) {
				for(int j = 0;j < n;j ++) {
					System.out.print(chs[i][j]);
				}
				System.out.println();
			}
			System.out.println();
			return;
		}
		for(int y = 0;y < n;y ++) {
			if(!col[y] && !dg[y - x + n] && !udg[y + x]) {
				chs[x][y] = 'Q';
				col[y] = dg[y - x + n] = udg[y + x] = true;
				dfs(x + 1);
				chs[x][y] = '.';
				col[y] = dg[y - x + n] = udg[y + x] = false;
			}
		}
		
	}
}
相关推荐
焦耳加热15 分钟前
阿德莱德大学Nat. Commun.:盐模板策略实现废弃塑料到单原子催化剂的高值转化,推动环境与能源催化应用
人工智能·算法·机器学习·能源·材料工程
wan5555cn23 分钟前
多张图片生成视频模型技术深度解析
人工智能·笔记·深度学习·算法·音视频
u6061 小时前
常用排序算法核心知识点梳理
算法·排序
蒋星熠3 小时前
Flutter跨平台工程实践与原理透视:从渲染引擎到高质产物
开发语言·python·算法·flutter·设计模式·性能优化·硬件工程
小欣加油3 小时前
leetcode 面试题01.02判定是否互为字符重排
数据结构·c++·算法·leetcode·职场和发展
3Cloudream4 小时前
LeetCode 003. 无重复字符的最长子串 - 滑动窗口与哈希表详解
算法·leetcode·字符串·双指针·滑动窗口·哈希表·中等
王璐WL4 小时前
【c++】c++第一课:命名空间
数据结构·c++·算法
空白到白4 小时前
机器学习-聚类
人工智能·算法·机器学习·聚类
索迪迈科技4 小时前
java后端工程师进修ing(研一版 || day40)
java·开发语言·学习·算法
zzzsde5 小时前
【数据结构】队列
数据结构·算法