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;
			}
		}
		
	}
}
相关推荐
聚客AI11 小时前
🙋‍♀️Transformer训练与推理全流程:从输入处理到输出生成
人工智能·算法·llm
大怪v13 小时前
前端:人工智能?我也会啊!来个花活,😎😎😎“自动驾驶”整起!
前端·javascript·算法
惯导马工15 小时前
【论文导读】ORB-SLAM3:An Accurate Open-Source Library for Visual, Visual-Inertial and
深度学习·算法
骑自行车的码农16 小时前
【React用到的一些算法】游标和栈
算法·react.js
博笙困了17 小时前
AcWing学习——双指针算法
c++·算法
moonlifesudo17 小时前
322:零钱兑换(三种方法)
算法
NAGNIP1 天前
大模型框架性能优化策略:延迟、吞吐量与成本权衡
算法
美团技术团队1 天前
LongCat-Flash:如何使用 SGLang 部署美团 Agentic 模型
人工智能·算法
Fanxt_Ja2 天前
【LeetCode】算法详解#15 ---环形链表II
数据结构·算法·leetcode·链表
侃侃_天下2 天前
最终的信号类
开发语言·c++·算法