拓扑排序入门

参考了这篇博客

经典写法:

vector findOrder(int n, vector& in, vector>& edges) {
    // in: 节点的入度表
    // edges:图的邻接表
    queue q;
    vector ans;
    for(int i = 0 ; i < n ; i++)
        if(in[i] == 0) q.push(i);
    while(!q.empty())
    {
        int p = q.front();
        q.pop();
        ans.push_back(p);
        for(int i = 0 ; i < edges[p].size() ; i++)
        {
            int y = edges[p][i];
            --in[y];
            if(in[y] == 0) q.push(y);
        }
    }
    if(ans.size() == n) return ans;
    else return {};
}

套用这个板子可以完美解决leetcode207.课程表和210.课程表II,只需要求一下入度表和邻接表即可。

你可能感兴趣的:(拓扑排序入门)