05.17leetcode每日一题-- 210.课程表II

这道题是一道很经典的拓扑排序问题

拓扑排序过程: 找到第一个入度为0的顶点,在图中去掉这个顶点,找下一个入度为0的顶点,直至图中没有顶点或者找不到入度为0的顶点为止。

变量：建立邻接链表. 一个统计结点入度的数组, 一个用于保存入度为0的队列.
算法思想:

1. 根据输入条件初始化邻接链表和入度数组.
2. 选定入度为0的顶点进队列.
3. 进入循环,循环终止条件为队列为空
4. 循环的动作是弹出队列头,将队列头作为弧起点的弧去掉,弧终点对应的入度–.如果入度为0,加入队列.

代码

//拓扑排序的一个求解问题。
//拓扑排序过程: 建立邻接链表. 一个统计结点入度的数组, 一个用于保存入度为0的队列.
//算法思想,1.根据输入条件初始化邻接链表和入度数组.
        // 2.选定入度为0的顶点进队列.
        // 3.进入循环,循环终止条件为队列为空
        // 4.循环的动作是弹出队列头,将队列头作为弧起点的弧去掉,弧终点对应的入度--.如果入度为0,加入队列.

#define max 10000
class Solution {
public:
    vector<int> findOrder(int numCourses, vector<vector<int>>& prerequisites) {
        vector<int> ret;
        vector<int> edge[max];
        queue<int> q;
        int inDegree[max];
        for(int i=0;i<max;i++){
            edge[i].clear();
            inDegree[i]=0;
        }
        //初始化邻接链表和入度数组.
        for(int i=0;i<prerequisites.size();i++){
                //[i][1]是弧的起点,[i][0]是弧的终点
                edge[prerequisites[i][1]].push_back(prerequisites[i][0]);
                //终点的入度++
                inDegree[prerequisites[i][0]]++;
            }
        //选定入度为0的顶点入队列.
        for(int i=0;i<numCourses;i++){
            if(inDegree[i]==0){
                q.push(i);
            }
        }
        //进入循环.
        while(!q.empty()){
            int cur=q.front();
            q.pop();
            ret.push_back(cur);
            //将队列头作为弧起点的弧去掉,弧终点对应的入度--,如果入度为0,加入队列.
            for(int i=0;i<edge[cur].size();i++){
                int tail=edge[cur][i];
                inDegree[tail]--;
                if(inDegree[tail]==0){
                    q.push(tail);
                }
            }
        }
        if(ret.size()==numCourses)
            return ret;
        else return {};
    }
};