Leetcode 210. 课程表 II

和Leetcode 207. 课程表一样,拓扑排序,上一个记录拓扑排序过程中的数量,这一个记录顺序

class Solution {
public:
    vector<int> findOrder(int numCourses, vector<pair<int, int>>& prerequisites) {
        vector<vector<int>> A(numCourses,vector<int>());
        vector<int> cnt(numCourses,0),ans;
        for(auto& x:prerequisites)
            A[x.second].push_back(x.first),++cnt[x.first];
        queue<int> q;
        for(int i=0;i<numCourses;++i)
            if(cnt[i]==0) q.push(i);
        while(!q.empty()){
            auto x=q.front();q.pop();
            ans.push_back(x);
            for(auto &y:A[x])
                if(--cnt[y]==0) q.push(y);
        }
        if(ans.size()!=numCourses) ans.clear();
        return ans;
    }
};

猜你喜欢

转载自blog.csdn.net/bendaai/article/details/81151874