1. 程式人生 > 其它 >210. 課程表 II(拓撲排序bfs, 與207一樣)

210. 課程表 II(拓撲排序bfs, 與207一樣)

 

難度中等

現在你總共有 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; } };