LeetCode 207 Course Schedule

LeetCode 207 Course Schedule

题意

有0~n-1共n门课你需要进修，其中一些课程需要在修完其他课程后才能进修，例如(0,1)表示要进修课程0你必须先修完课程1。给出课程的总数以及一个列表表示所有的先决条件，问是否可能修完所有课程。
注：

• 课程先决条件可以看作是由一个图的边(edge)组成
• 不会出现重复的边

思路

假如将每个先决条件(a,b)看作是b到a的有向边，整个课程实际上是个有向图，那么是否能修完所有课程就是判断该有向图是否存在一个环。由于给出了所有的边，因此可以通过邻接表的方式构造图。

方法一

深度优先搜索图，搜索过程中如果出现重复访问的点则说明存在环，如下图dfs时在第二次访问到d时可知道该图包含了环。

a ->  b <- e
      ↓    ↑
      c -> d

方法二

通过不断地删除边来判断是否有环。对于一个无环有向图来说，从起点(即入度为0的节点)开始每次向前走一步并删除走过的边，最终一定会走到终点且图中不再含有任何边 - 所有节点入度为0。具体过程如下:

1. 找出所有入度为0的节点加入队列Q
2. 每次从Q中取出一个节点u，删除从u出发的所有边 - 即这些边指向的节点入度减1
3. 若出现了新的度为0的节点v，则将v也加入Q中
4. 重复上述操作直到队列位空
5. 若此时仍有节点的入度不为0，则说明图中有环
   需要注意的是若图中不包含入度为0的节点，则该图一定包含环。

代码

深搜的代码较为简单，只贴第二种方法的代码（其实更简单）

typedef pair<int, int> PII;
class Solution {
  public:
    bool canFinish(int numCourses, vector<pair<int, int>> &prerequisites) {
        if (prerequisites.size() <= 1)
            return true;
        vector<int> g[numCourses];
        int inDegree[numCourses] = {};
        for (PII tmppair : prerequisites) {
            g[tmppair.second].push_back(tmppair.first);
            inDegree[tmppair.first]++;
        }
        queue<int> q;
        for (int i = 0; i < numCourses; i++)
            if (inDegree[i] == 0)
                q.push(i);
        while (!q.empty()) {
            int u = q.front(); q.pop();
            for (int v : g[u])
                if (--inDegree[v] == 0)
                    q.push(v);
        }
        for (int i = 0; i < numCourses; i++)
            if (inDegree[i] != 0)
                return false;
        return true;
    }
};

总结

深度优先搜索可能会重复搜索已经搜索过的点，而删边的方法则不会出现此问题。