LeetCode 207. Course Schedule

这段时间课上在讲图论，就在LeetCode上找题做，图这部分有很多算法，最基本的还是DFS和BFS，下面是其中一道关于选课的题。
有n门课程，代号分别为0，1，2……n-1。其中有某些课程需要选过另一些课程才能选，如要选课程1，必须先选课程0和课程2。问能否把课程都修完。
选课的问题和穿衣服的问题类似，是一个典型的拓扑排序的问题，使用拓扑排序的方法不仅能给出是否可以修完的判断，还可以给出选课的顺序，代码如下：

#include <iostream>
#include <vector>
#include <set>

using namespace std;

class Solution {
public:
    bool canFinish(int numCourses, vector<pair<int, int> >& prerequisites) {
        unsigned int n = (unsigned int) numCourses;
        unsigned int i;

        vector<set<int> > G(n);
        vector<int> inDegree(n, 0);
        vector<int> topoSort;

        for (i = 0; i < prerequisites.size(); ++i) {
            G[prerequisites[i].second].insert(prerequisites[i].first);
            inDegree[prerequisites[i].first]++;
        }

        do {
            for (i = 0; i < n; ++i) {
                if (inDegree[i] == 0) {

                    inDegree[i]--;
                    topoSort.push_back(i);

                    for (set<int>::iterator it = G[i].begin(); it != G[i].end(); ++it)
                        inDegree[*it]--;

                    break;
                }
            }
        } while (i < n && topoSort.size() < n);

        return topoSort.size() == n;
    }

};

另一方面，本题不需要求出选课的顺序，也可以理解为检测一个图中是否存在环，若存在环，则有某些课互为因果，无法修完。
可以用DFS来检测，我们知道在深搜中，当访问一个节点时，其祖先一定没有结束访问（被它阻塞），所以祖先节点一定在栈中，如果这个节点的子节点中有一个是其祖先，那么一定会形成一个环。
DFS中需要的栈一般用递归来实现，可以开一个辅助的vector onStack，记录节点course是否在栈中，若course入栈，则将onStack[course]设为true，出栈改回false。在DFS时，如果试图再次访问仍在栈中的点，则说明存在环，返回false，若遍历所有节点没有这种情况，则返回true。

#include <iostream>
#include <vector>
#include <map>
#include <set>
#include <algorithm>
#include <utility>

using namespace std;

class Solution {
    vector<set<int> > graph;
    vector<int> visited;
    vector<int> onStack;
    bool ret;

public:
    bool canFinish(int numCourses, vector<pair<int, int> >& prerequisites) {
        ret = true;

        graph.resize(numCourses);
        for (unsigned int i = 0; i < prerequisites.size(); ++i)
            graph[prerequisites[i].second].insert(prerequisites[i].first);

        visited.resize(numCourses, false);
        onStack.resize(numCourses, false);

        for (int i = 0; i < numCourses; ++i) {
            if (!visited[i])
                dfs(i);

            if (ret == false)
                return ret;
        }

        return ret;
    }

    void dfs(int course) {
        visited[course] = true;
        onStack[course] = true;

        for (set<int>::iterator it = graph[course].begin(); it != graph[course].end();
                ++it) {
            if (onStack[*it]) {
                ret = false;
                return;
            } else if (!visited[*it])
                dfs(*it);
        }

        onStack[course] = false;
    }
};

总结来讲，感觉图论的题首先要将问题转化，按需要构建出一个合适的图，然后从图论中的基本算法出发，寻求解决方法。