leetcode——210. 课程表 II

思路

1. 建立有向图，记录节点入度
2. 将入度为0的结点放入队列
3. 进行广度优先搜索，将队头放入答案
4. 相邻结点入度-1，入度更新为0则放入队列

代码

class Solution {
private:
    // 存储有向图
    vector<vector<int>> edges;
    // 存储每个节点的入度
    vector<int> indeg;
    // 存储答案
    vector<int> result;

public:
    vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
        edges.resize(numCourses);
        indeg.resize(numCourses);
        //建图 记录每个结点的入度
        for (const auto& info: prerequisites) {
            edges[info[1]].push_back(info[0]);
            ++indeg[info[0]];
        }

        // 将所有入度为 0 的节点放入队列中
        queue<int> q;
        for (int i = 0; i < numCourses; ++i) 
            if (indeg[i] == 0)
                q.push(i);
            
        //广度优先搜索
        while (!q.empty()) {
            int u = q.front();q.pop();
            result.emplace_back(u);
            for (int v: edges[u]) {
                --indeg[v];
                // 如果相邻节点 v 的入度为 0，就可以选 v 对应的课程了
                if (indeg[v] == 0) 
                    q.push(v);
            }
        }

        if(result.size()==numCourses)
            return result;
        else
            return {};
    }
};