LeetCode题解–207. Course Schedule

xiaoxiao2021-02-28  70

链接

LeetCode题目:https://leetcode.com/problems/course-schedule/

难度:Medium

题目

There are a total of n courses you have to take, labeled from 0 to n - 1. Some courses may have prerequisites, for example to take course 0 you have to first take course 1, which is expressed as a pair: [0,1] Given the total number of courses and a list of prerequisite pairs, is it possible for you to finish all courses?

题目大意是选n门课程,某些课程可能有先修课程,试问能否成功选择全部课程。

分析

这题是一道典型的拓扑排序问题,拓扑排序的定义是:将有向图中的顶点以线性方式进行排序,对于任何连接自顶点u到顶点v的有向边uv,在最后的排序结果中,顶点u总是在顶点v的前面。 典型的实现算法有Kahn算法和基于DFS的算法,Kahn算法相比后者的好处是不需要检测图为DAG,所以我选择了Kahn算法。 做法是维护一个入度为0的点的集合(这里选择了队列),每次从集合中取出一个点n,依次删除从n到m的边,如果删后m的入度为0,则把点m也放入这个集合。最后集合为空而且不存在边时,证明选课成功。时间复杂度是O(V+E),其中V为边数,E为点数。

代码

class Solution { public: bool canFinish(int numCourses, vector<pair<int, int>> &prerequisites) { edgeNum = prerequisites.size(); inDegrees = vector<int>(numCourses, 0); for (int i = 0; i < numCourses; i++) { edges.push_back(vector<int>()); } for (auto pre:prerequisites) { edges[pre.second].push_back(pre.first); inDegrees[pre.first]++; } for (int i = 0; i < inDegrees.size(); i++) { if (inDegrees[i] == 0) points.push(i); } while (points.size()) { int n = points.front(); points.pop(); for (auto m:edges[n]) { edgeNum--; inDegrees[m]--; if (!inDegrees[m]) points.push(m); } } return edgeNum == 0; } private: int edgeNum; vector<int> inDegrees; vector<vector<int>> edges; queue<int> points; };
转载请注明原文地址: https://www.6miu.com/read-59276.html

最新回复(0)