LeetCode 210.课程表II

主页有其他数据结构内容(持续更新中)

难度:Medium

代码:

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 (auto info: prerequisites) {
            //  info[1]指向info[0]
            edges[info[1]].emplace_back(info[0]);
            indeg[info[0]]++;
        }
        queue<int> q;
        //  将所有入度为0的节点放入队列中
        for (int i = 0; i < numCourses; ++i) {
            if (indeg[i] == 0) {
                q.push(i);
            }
        }
        //  bfs
        while (!q.empty()) {
            //  从队首取出一个节点
            int temp = q.front();
            q.pop();
            //  把该节点放入答案数组中
            result.emplace_back(temp);
            for (auto c: edges[temp]) {
                indeg[c]--;
                //  如果相邻节点c的入度为 0,就把c放入队列,即可以选c对应的课程了
                if (indeg[c] == 0) {
                    q.push(c);
                }
            }
        }
        if (result.size() != numCourses) {
            //  表明有回路,无法进行拓扑排序
            return {};
        }
        return result;
    }
};

你可能感兴趣的:(LeetCode,(C++,Version),leetcode,算法,数据结构)