Code & Func

第52天。

昨晚去见了一下一起做东西的研究生师兄师姐们，感觉很nice,不嫌弃我什么都不会还教了我很多东西，恩，等下开始学机器学习的东西。

There are a total of n courses you have to take, labeled from 0 to n - 1.

Some courses may have prerequisites, for example to take course 0 you have to first take course 1, which is expressed as a pair: [0,1]

Given the total number of courses and a list of prerequisite pairs, is it possible for you to finish all courses?

For example:

2, [[1,0]] There are a total of 2 courses to take. To take course 1 you should have finished course 0. So it is possible.

2, [[1,0],[0,1]] There are a total of 2 courses to take. To take course 1 you should have finished course 0, and to take course 0 you should also have finished course 1. So it is impossible. Note: The input prerequisites is a graph represented by a list of edges, not adjacency matrices. Read more about how a graph is represented. You may assume that there are no duplicate edges in the input prerequisites.

这个问题可以转化成——有向图是否有环路。

这里是使用拓扑排序做的. 拓扑排序是只能在有向无环图中进行排序，如果它有环，那么它进会出错，我们对这个图进行一次拓扑排序就可以知道这个图是不是有环了。

一开始是使用维护一个入度的数组，然后通过不断删除入度为0点的方式来完成拓扑排序的，但是超时了，所以这里用DFS的方法来实现，而且这种方法还比之前的要简单。

原本的DFS需要一个visited，来表示某个节点是否被访问了，这里扩展一下visited，原本的visited只有两个状态：被访问了，未被访问。这里加入一个新的状态访问中,这里用-1来表示。

之所以要加入这个状态，是因为我们需要判断这个图是否有环路。让我们看个例子。

我们尝试着对这个图进行一次DFS:

1
2
2
3->8->9->10
3
5->11
4
7

从上面我们可以知道这个图，需要四次调用DFS的递归函数才能完成整个遍历，我们认为如果某个节点在某次递归中，那么它的状态就是访问中,也就是说在第二次调用DFS的递归函数时如果访问了3,8节点，准备访问9节点时，3,8就是被访问状态，一旦访问完所以节点（也就是访问完10,这是在这条链路中所以节点都被访问了，递归函数开始返回，然后我们可以依次把10,9,8,3的设为已访问的状态。

讲了那么多，如果定义访问中状态，好像还没有提到他的用处，还是刚才的例子，如果上图加上一个9->3的边，那么我们是不是在访问9时，发现他可以通向一个访问中的节点（即3节点)，这时说明他们之间必定有回路。

大概的思路就是这样吧，其实看代码会简单一点：

1
def helper(self,v,visited,graph):
2
    visited[v] = -1 #设置为访问中
3
    for i in graph[v]:
4
        if visited[i] == -1: #访问中
5
            return False
6
        if visited[i] == 0: #未访问
7
            if self.helper(i,visited,graph) == False:
8
                return False
9
    visited[v] = 1 # 递归函数开始返回了，设置为已访问的状态
10
    return True
11
def canFinish(self, numCourses, prerequisites):
12
    """
13
    :type numCourses: int
14
    :type prerequisites: List[List[int]]
15
    :rtype: bool
14 collapsed lines
16
    """
17
    visited = [0]*numCourses #初始时，所有节点都未访问
18
    #将边集转化成邻接表
19
    graph = [[] for i in range(numCourses)]
20
    for e in prerequisites:
21
        graph[e[0]].append(e[1])
22

23
    #DFS
24
    for i in range(numCourses):
25
        if visited[i] == False:
26
            if self.helper(i,visited,graph) == False:
27
                return False
28

29
    return True

dicuss中的BFS解法：

1
class Solution {
2
public:
3
    bool canFinish(int numCourses, vector<pair<int, int>>& prerequisites) {
4
        vector<unordered_set<int>> graph = make_graph(numCourses, prerequisites);
5
        vector<int> degrees = compute_indegree(graph);
6
        for (int i = 0; i < numCourses; i++) {
7
            int j = 0;
8
            for (; j < numCourses; j++)
9
                if (!degrees[j]) break;
10
            if (j == numCourses) return false;
11
            degrees[j] = -1;
12
            for (int neigh : graph[j])
13
                degrees[neigh]--;
14
        }
15
        return true;
16 collapsed lines
16
    }
17
private:
18
    vector<unordered_set<int>> make_graph(int numCourses, vector<pair<int, int>>& prerequisites) {
19
        vector<unordered_set<int>> graph(numCourses);
20
        for (auto pre : prerequisites)
21
            graph[pre.second].insert(pre.first);
22
        return graph;
23
    }
24
    vector<int> compute_indegree(vector<unordered_set<int>>& graph) {
25
        vector<int> degrees(graph.size(), 0);
26
        for (auto neighbors : graph)
27
            for (int neigh : neighbors)
28
                degrees[neigh]++;
29
        return degrees;
30
    }
31
};

好像就是我一开始做的那种想法，但是为什么我的又没通过。