每一个课程看作一个点,先修课程连出一条边指向后续课程,整体形成一个图。我们需要对这个图进行拓扑排序,如果图中存在环,则不存在拓扑序。拓扑排序最直接的方法是BFS。时间复杂度是O(n + m)
class Solution { private: // 存储有向图 vector<vector<int>> edges; // 存储每个节点的入度 vector<int> indeg; // 存储答案 vector<int> res; public: vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) { edges.resize(numCourses); indeg.resize(numCourses); for (auto &prerequisite : prerequisites) { edges[prerequisite[1]].push_back(prerequisite[0]); indeg[prerequisite[0]]++; } queue<int> q; for (int i = 0; i < numCourses; i++) { if (indeg[i] == 0) { q.push(i); } } while (!q.empty()) { auto vertex = q.front(); q.pop(); if (indeg[vertex] == 0) { res.push_back(vertex); } for (auto v : edges[vertex]) { indeg[v]--; if (indeg[v] == 0) { q.push(v); } } } vector<int> null; return (res.size() != numCourses ? null : res); } };