210. 課程表 II(拓撲排序bfs, 與207一樣)
阿新 • • 發佈:2022-05-11
難度中等
現在你總共有 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]
。
class Solution { public: vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) { unordered_map<int,std::vector<int>> map; vector<int> ingreed(numCourses,0); for(int i = 0; i < prerequisites.size();i++) { ingreed[prerequisites[i][0]]++; map[prerequisites[i][1]].emplace_back(prerequisites[i][0]); } queue<int> q;for(int i = 0; i < ingreed.size();i++) { if (ingreed[i]==0) { q.push(i); } } std::vector<int> res; while(!q.empty()) { int top = q.front(); q.pop(); res.emplace_back(top); for(auto adj: map[top]) { ingreed[adj]--; if (ingreed[adj]==0) { q.push(adj); } } } //cout << res.size() << endl; if (res.size()!=numCourses){ res.clear(); } return res; } };