Course Schedule II

    xiaoxiao2022-07-02  106

    原题链接 这道题就是典型的拓扑排序问题。简单说一下拓扑排序:对一个有向无环图G进行拓扑排序,是将G中所有顶点排成一个线性序列,使得图中任意一对顶点u和v,若边(u,v)∈E(G),则u在线性序列中出现在v之前。

    拓扑排序算法主要是循环执行以下两步,直到不存在入度为0的顶点为止。 (1) 选择一个入度为0的顶点并输出之; (2) 从网中删除此顶点及所有出边。 循环结束后,若输出的顶点数小于网中的顶点数,则说明存在回路,

    回到这道题中,首先需要遍历一次pre中所有的有向边,统计每一个点的入度,为此我使用了map;并且记录每一个起始点的邻接点。 先将入度为0的节点进栈,然后进行广度优先遍历,将栈顶的元素保存到栈res中,并依次将栈顶的邻接点的入度减1,如果邻接点的入度为0,那就将邻接点压入栈中,直到栈空。

    最后判断res的长度是否等于n,如果相等,那就说明有向图中不存在回路,否则存在。

    代码如下:

    class Solution { public: vector<int> findOrder(int n, vector<vector<int>>& pre) { vector<int> res,tmpres; vector<int> degree; vector<set<int>> neighbor(n,set<int>()); for(auto item:pre){ degree[item[0]]++; neighbor[item[1]].insert(item[0]); } for(int i=0;i<n;i++){ if(degree[i]==0){ tmpres.push_back(i); } } while(tmpres.size()){ int t = tmpres.back();tmpres.pop_back(); res.push_back(t); for(auto child:neighbor[t]){ degree[child]--; if(degree[child]==0)tmpres.push_back(child); } } return res.size()==n?res:vector<int>(); } };
    最新回复(0)