实习面试算法准备之图论

ops/2024/9/22 23:05:21/

这里写目录标题

  • 1 基础内容
    • 1.1 图的表示
    • 1.2图的遍历
  • 2 例题
    • 2.1 所有可能的路径
    • 2.2 课程表(环检测算法
    • 2.3 课程表 II (拓扑排序算法
      • 2.3.1 拓扑排序 DFS版

1 基础内容

图没啥高深的,本质上就是个高级点的多叉树而已,适用于树的 DFS/BFS 遍历算法,全部适用于图。

1.1 图的表示

图的存储在算法题中常用邻接表和邻接矩阵表示:
在这里插入图片描述在这里插入图片描述

java">// 邻接表
// graph[x] 存储 x 的所有邻居节点
List<Integer>[] graph;// 邻接矩阵
// matrix[x][y] 记录 x 是否有一条指向 y 的边
boolean[][] matrix;

有向加权图怎么实现?很简单呀:
如果是邻接表,我们不仅仅存储某个节点 x 的所有邻居节点,还存储 x 到每个邻居的权重,不就实现加权有向图了吗?
如果是邻接矩阵,matrix[x][y] 不再是布尔值,而是一个 int 值,0 表示没有连接,其他值表示权重,不就变成加权有向图了吗?
如果用代码的形式来表现,大概长这样:

java">// 邻接表
// graph[x] 存储 x 的所有邻居节点以及对应的权重
List<int[]>[] graph;// 邻接矩阵
// matrix[x][y] 记录 x 指向 y 的边的权重,0 表示不相邻
int[][] matrix;

1.2图的遍历

图怎么遍历?还是那句话,参考多叉树,多叉树的 DFS 遍历框架如下:

java">/* 多叉树遍历框架 */
void traverse(TreeNode root) {if (root == null) return;// 前序位置for (TreeNode child : root.children) {traverse(child);}// 后序位置
}

图和多叉树最大的区别是,图是可能包含环的,你从图的某一个节点开始遍历,有可能走了一圈又回到这个节点,而树不会出现这种情况,从某个节点出发必然走到叶子节点,绝不可能回到它自身。

所以,如果图包含环,遍历框架就要一个 visited 数组进行辅助:

java">// 记录被遍历过的节点
boolean[] visited;
// 记录在一次traverse中递归过的结点
boolean[] onPath;/* 图遍历框架 */
void traverse(Graph graph, int s) {if (visited[s]) return;// 经过节点 s,标记为已遍历visited[s] = true;// 做选择:标记节点 s 在路径上onPath[s] = true;for (int neighbor : graph.neighbors(s)) {traverse(graph, neighbor);}// 撤销选择:节点 s 离开路径onPath[s] = false;
}

注意 visited 数组和 onPath 数组的区别
类比贪吃蛇游戏,visited 记录蛇经过过的格子,而 onPath 仅仅记录蛇身。在图的遍历过程中,onPath 用于判断是否成环,类比当贪吃蛇自己咬到自己(成环)的场景。
如果让你处理路径相关的问题,这个 onPath 变量是肯定会被用到的,比如 拓扑排序 中就有运用。
这个 onPath 数组的操作很像前文 回溯算法核心套路 中做「做选择」和「撤销选择」,区别在于位置:回溯算法的「做选择」和「撤销选择」在 for 循环里面,而对 onPath 数组的操作在 for 循环外面。

回忆:
对于回溯算法,我们需要在「树枝」上做选择和撤销选择:
在这里插入图片描述
反映到代码上就是:

java">// DFS 算法,关注点在节点
void traverse(TreeNode root) {if (root == null) return;printf("进入节点 %s", root);for (TreeNode child : root.children) {traverse(child);}printf("离开节点 %s", root);
}// 回溯算法,关注点在树枝
void backtrack(TreeNode root) {if (root == null) return;for (TreeNode child : root.children) {// 做选择printf("从 %s 到 %s", root, child);backtrack(child);// 撤销选择printf("从 %s 到 %s", child, root);}
}

另一种解释就是,如果用回溯的方法遍历树,你会发现根节点被漏掉了:

java">void traverse(TreeNode root) {if (root == null) return;for (TreeNode child : root.children) {printf("进入节点 %s", child);traverse(child);printf("离开节点 %s", child);}
}

所以对于这里「图」的遍历,我们应该用 DFS 算法,即把 onPath 的操作放到 for 循环外面,否则会漏掉记录起始点的遍历。
说了这么多 onPath 数组,再说下 visited 数组,其目的很明显了,由于图可能含有环,visited 数组就是防止递归重复遍历同一个节点进入死循环的。

当然,如果题目告诉你图中不含环,可以把 visited 数组都省掉,基本就是多叉树的遍历。

2 例题

2.1 所有可能的路径

给你一个有 n 个节点的 有向无环图(DAG),请你找出所有从节点 0 到节点 n-1 的路径并输出(不要求按特定顺序)
graph[i] 是一个从节点 i 可以访问的所有节点的列表(即从节点 i 到节点 graph[i][j]存在一条有向边)

示例1:
在这里插入图片描述
输入:graph = [[1,2],[3],[3],[]]
输出:[[0,1,3],[0,2,3]]
解释:有两条路径 0 -> 1 -> 3 和 0 -> 2 -> 3

代码以及思路:
解法很简单,以 0 为起点遍历图,同时记录遍历过的路径,当遍历到终点时将路径记录下来即可。
既然输入的图是无环的,我们就不需要 visited 数组辅助了,直接套用图的遍历框架:

java">class Solution {List<List<Integer>> res = new ArrayList();public List<List<Integer>> allPathsSourceTarget(int[][] graph) {List<Integer> path = new ArrayList();traverse(graph,0,path);return res;}public void traverse(int[][] graph,int s,List<Integer> path){path.add(s);int n = graph.length;if(s == n-1){res.add(new ArrayList(path));}for(int i:graph[s]){traverse(graph,i,path);}path.removeLast();}
}

2.2 课程表(环检测算法

你这个学期必须选修 numCourses 门课程,记为 0 到 numCourses - 1 。
在选修某些课程之前需要一些先修课程。 先修课程按数组 prerequisites 给出,其中 prerequisites[i] = [ai, bi] ,表示如果要学习课程 ai 则 必须 先学习课程 bi 。
例如,先修课程对 [0, 1] 表示:想要学习课程 0 ,你需要先完成课程 1 。
请你判断是否可能完成所有课程的学习?如果可以,返回 true ;否则,返回 false 。

示例 1:
输入:numCourses = 2, prerequisites = [[1,0]]
输出:true
解释:总共有 2 门课程。学习课程 1 之前,你需要完成课程 0 。这是可能的。
示例 2:
输入:numCourses = 2, prerequisites = [[1,0],[0,1]]
输出:false
解释:总共有 2 门课程。学习课程 1 之前,你需要先完成​课程 0 ;并且学习课程 0 之前,你还应先完成课程 1 。这是不可能的。

2.2.1 环检测算法 DFS版

看到依赖问题,首先想到的就是把问题转化成「有向图」这种数据结构,只要图中存在环,那就说明存在循环依赖。
具体来说,我们首先可以把课程看成「有向图」中的节点,节点编号分别是 0, 1, …, numCourses-1,把课程之间的依赖关系看做节点之间的有向边。
比如说必须修完课程 1 才能去修课程 3,那么就有一条有向边从节点 1 指向 3。
所以我们可以根据题目输入的 prerequisites 数组生成一幅类似这样的图:
在这里插入图片描述
如果发现这幅有向图中存在环,那就说明课程之间存在循环依赖,肯定没办法全部上完;反之,如果没有环,那么肯定能上完全部课程。
好,那么想解决这个问题,首先我们要把题目的输入转化成一幅有向图,然后再判断图中是否存在环。
怎么转化为图?以刷题的经验,大概率是要转化为邻接表:

java"># graph[s] 是一个列表,存储着节点 s 所指向的节点。
List<Integer>[] graph;

首先可以写一个建图函数:

java">List<Integer>[] buildGraph(int numCourses, int[][] prerequisites) {// 图中共有 numCourses 个节点List<Integer>[] graph = new LinkedList[numCourses];for (int i = 0; i < numCourses; i++) {graph[i] = new LinkedList<>();}for (int[] edge : prerequisites) {int from = edge[1], to = edge[0];// 添加一条从 from 指向 to 的有向边// 边的方向是「被依赖」关系,即修完课程 from 才能修课程 tograph[from].add(to);}return graph;
}

图建出来了,怎么判断图中有没有环呢?
先不要急,我们先来思考如何遍历这幅图,只要会遍历,就可以判断图中是否存在环了。

java">// 防止重复遍历同一个节点
boolean[] visited;boolean canFinish(int numCourses, int[][] prerequisites) {List<Integer>[] graph = buildGraph(numCourses, prerequisites);visited = new boolean[numCourses];for (int i = 0; i < numCourses; i++) {traverse(graph, i);}
}void traverse(List<Integer>[] graph, int s) {// 代码见上文
}

注意图中并不是所有节点都相连,所以要用一个 for 循环将所有节点都作为起点调用一次 DFS 搜索算法
这样,就能遍历这幅图中的所有节点了,你打印一下 visited 数组,应该全是 true。
现在可以思考如何判断这幅图中是否存在环。
你也可以把 traverse 看做在图中节点上游走的指针,只需要再添加一个布尔数组 onPath 记录当前 traverse 经过的路径:

java">boolean[] onPath;
boolean[] visited;boolean hasCycle = false;void traverse(List<Integer>[] graph, int s) {if (onPath[s]) {// 发现环!!!hasCycle = true;}if (visited[s] || hasCycle) {return;}// 将节点 s 标记为已遍历visited[s] = true;// 开始遍历节点 sonPath[s] = true;for (int t : graph[s]) {traverse(graph, t);}// 节点 s 遍历完成onPath[s] = false;
}

这里就有点回溯算法的味道了,在进入节点 s 的时候将 onPath[s] 标记为 true,离开时标记回 false,如果发现 onPath[s] 已经被标记,说明出现了环。

注意 visited 数组和 onPath 数组的区别,因为二叉树算是特殊的图,所以用遍历二叉树的过程来理解下这两个数组的区别:
在这里插入图片描述
上述 GIF 描述了递归遍历二叉树的过程,在 visited 中被标记为 true 的节点用灰色表示,在 onPath 中被标记为 true 的节点用绿色表示。
因此,整理一下,完整代码如下:

java">class Solution {boolean[] onPath;boolean[] visited;//false表示没有环boolean result = false;public boolean canFinish(int numCourses, int[][] prerequisites) {//建图List<Integer>[] graph = buildGraph(numCourses,prerequisites);//遍历onPath = new boolean[numCourses];visited = new boolean[numCourses];for(int i = 0;i<numCourses;i++){//遍历每个结点traverse(graph,i);}return !result;}public List<Integer>[] buildGraph(int numCourses, int[][] prerequisites){List<Integer>[] graph = new ArrayList[numCourses];//图有numCourse个结点for(int i = 0;i<numCourses;i++){graph[i] = new ArrayList();}for(int[] i:prerequisites){int start = i[0];int end = i[1];graph[start].add(end);}return graph;}public void traverse(List<Integer>[] graph,int s){if(onPath[s]){result = true;}if(visited[s] || result){return;}visited[s] = true;onPath[s] = true;for(int i:graph[s]){traverse(graph,i);}onPath[s] = false;}
}

2.2.2 环检测算法 BFS版

2.3 课程表 II (拓扑排序算法

现在你总共有 numCourses 门课需要选,记为 0 到 numCourses - 1。给你一个数组 prerequisites ,其中 prerequisites[i] = [ai, bi] ,表示在选修课程 ai 前 必须 先选修 bi 。
例如,想要学习课程 0 ,你需要先完成课程 1 ,我们用一个匹配来表示:[0,1] 。
返回你为了学完所有课程所安排的学习顺序。可能会有多个正确的顺序,你只要返回 任意一种 就可以了。如果不可能完成所有课程,返回 一个空数组 。

示例 1:

输入:numCourses = 2, prerequisites = [[1,0]]
输出:[0,1]
解释:总共有 2 门课程。要学习课程 1,你需要先完成课程 0。因此,正确的课程顺序为 [0,1] 。
示例 2:

输入:numCourses = 4, prerequisites = [[1,0],[2,0],[3,1],[3,2]]
输出:[0,2,1,3]
解释:总共有 4 门课程。要学习课程 3,你应该先完成课程 1 和课程 2。并且课程 1 和课程 2 都应该排在课程 0 之后。
因此,一个正确的课程顺序是 [0,1,2,3] 。另一个正确的排序是 [0,2,1,3] 。
示例 3:

输入:numCourses = 1, prerequisites = []
输出:[0]

2.3.1 拓扑排序 DFS版

什么是拓扑排序?
直观地说就是,让你把一幅图「拉平」,而且这个「拉平」的图里面,所有箭头方向都是一致的
很显然,如果一幅有向图中存在环,是无法进行拓扑排序的,因为肯定做不到所有箭头方向一致;反过来,如果一幅图是「有向无环图」,那么一定可以进行拓扑排序。
其实也不难看出来,如果把课程抽象成节点,课程之间的依赖关系抽象成有向边,那么这幅图的拓扑排序结果就是上课顺序。
首先,我们先判断一下题目输入的课程依赖是否成环,成环的话是无法进行拓扑排序的,所以我们可以复用上一道题的主函数:

java">public int[] findOrder(int numCourses, int[][] prerequisites) {if (!canFinish(numCourses, prerequisites)) {// 不可能完成所有课程return new int[]{};}// ...
}

那么关键问题来了,如何进行拓扑排序?是不是又要秀什么高大上的技巧了?
其实特别简单,将后序遍历的结果进行反转,就是拓扑排序的结果。
直接上代码:

java">class Solution {boolean[] onPath;boolean[] visited;boolean hasCycle = false;// 记录后序遍历结果List<Integer> postorder = new ArrayList<>();public int[] findOrder(int numCourses, int[][] prerequisites) {List<Integer>[] graph = buildgrap(numCourses,prerequisites);onPath = new boolean[numCourses];visited = new boolean[numCourses];for(int i = 0;i<numCourses;i++){traverse(graph,i);}if(hasCycle){return new int[]{};}Collections.reverse(postorder);int[] res = new int[numCourses];for(int i = 0;i<numCourses;i++){res[i] = postorder.get(i);}return res;}public List<Integer>[] buildgrap(int numCourses,int[][] prerequisites){List<Integer>[] graph = new ArrayList[numCourses];for(int i = 0;i<numCourses;i++){graph[i] = new ArrayList();}for(int[] i:prerequisites){int start = i[1];int end = i[0];graph[start].add(end);}return graph;}public void traverse(List<Integer>[] graph,int s){if(onPath[s]){hasCycle = true;}if(hasCycle || visited[s]){return;}onPath[s] = true;visited[s] = true;for(int i:graph[s]){traverse(graph,i);}onPath[s] = false;postorder.add(s);}
}

http://www.ppmy.cn/ops/28542.html

相关文章

xLua背包实践

准备工作 环境&#xff0c;代码 在C#代码方面我们需要准备单例模式基类&#xff0c;AB包管理器&#xff0c;lua解析器管理器 详情请见AB包管理器 xlua详解 然后是Xlua包和AB包&#xff0c;具体导入方法也在上面的链接中 然后是lua的三个文件 具体代码&#xff1a; JsonUtil…

Java:Thread类及常见方法大全(画图+源码详解)

Thread 类是 JVM 用来管理线程的一个类&#xff0c;每一个线程都有一个唯一的 Thread 类与之关联。Java中通常使用 Thread类来进行线程调度&#xff0c;线程管理。 目录 一、Thread 的常见构造方法 二、Thread 的几个常见属性 理解线程是否存活&#xff1a; 理解前台线程与…

斐波那契数列

&#x1f600;前言 斐波那契数列作为经典的数学问题&#xff0c;在计算机领域有着广泛的应用和研究价值。本文将探讨如何高效地求解斐波那契数列的第 n 项&#xff0c;通过不同的算法实现&#xff0c;并分析它们的时间复杂度和空间复杂度。 &#x1f3e0;个人主页&#xff1a;尘…

Gson打印按照想要的key顺序

默认大家都知道这个吧&#xff1f; val gson GsonBuilder().setPrettyPrinting().create() log(gson.toJson(bean))它是用于将对象bean&#xff0c;转成json以后&#xff0c;能够比较漂亮的打印出json的结构。我常用的是如下4个函数。 //就是jsonStr&#xff0c;使用该函数来…

翻译插件Translation和AndroidLocalize

目录 前言一、Translation二、AndroidLocalize 前言 两个Android Studio的插件&#xff0c;一个是Translation&#xff0c;一个是AndroidLocalize&#xff0c;前者是自动化翻译插件&#xff0c;可以选中代码中的代码进行翻译&#xff0c;也可在线进行查询翻译&#xff1b;后者…

【Linux】线程的创建、回收分离以及线程的同步互斥

目录 一、多线程的基本编程 二、线程安全&#xff08;同步互斥&#xff09; 1.使用互斥锁达到互斥 2.使用互斥锁和条件变量达到线程间的同步互斥 一、多线程的基本编程 线程回收&#xff1a;线程在运行时需要分配内存空间、处理器时间等系统资源&#xff0c;这些资源在线程…

C++每日一练——两个数组的交集

给定两个数组 nums1 和 nums2 &#xff0c;返回 它们的 交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。 示例 1&#xff1a; 输入&#xff1a;nums1 [1,2,2,1], nums2 [2,2] 输出&#xff1a;[2]示例 2&#xff1a; 输入&#xff1a;nums…

【设计模式】16、state 状态模式

文章目录 十六、state 状态模式16.1 自动购物机16.1.1 vending_machine_test.go16.1.2 vending_maching.go16.1.3 state.go16.1.4 no_good_state.go16.1.5 has_good_state.go 16.2 player16.2.1 player_test.go16.2.2 player.go16.2.3 state.go16.2.4 stopped_state.go16.2.5 p…