題目
你這個學期必須選修 numCourses 門課程,記為 0 到 numCourses - 1 。
在選修某些課程之前需要一些先修課程。 先修課程按數組 prerequisites 給出,其中 prerequisites[i] = [ai, bi] ,表示如果要學習課程 ai 則 必須 先學習課程 bi 。
例如,先修課程對 [0, 1] 表示:想要學習課程 0 ,你需要先完成課程 1 。
請你判斷是否可能完成所有課程的學習?如果可以,返回 true ;否則,返回 false 。
示例
輸入:numCourses = 2, prerequisites = [[1,0]]
輸出:true
解釋:總共有 2 門課程。學習課程 1 之前,你需要完成課程 0 。這是可能的。
解析
這道題首先主要的思路是用bfs來寫,考慮有如下數據:
n = 6,先決條件表:[[3, 0], [3, 1], [4, 1], [4, 2], [5, 3], [5, 4]]
對于這種題要構造兩個數據:
- 每個節點的入度數量
- 所有的節點構成一張圖,可以用map來表示,每個節點指向了哪些節點,這些節點用一個數組來表示
在此基礎上,不斷的將入度為0的節點放到隊列中消費掉,消費的時候看哪些節點的入度變成了0,則可以加入到隊列中,直到處理完成。
func canFinish(numCourses int, prerequisites [][]int) bool {var (edges = make(map[int][]int, numCourses) // 邊,也叫鄰接表,存的是每個位置,可以指向后面的哪些位置indeg = make([]int, numCourses) // 入度數組result []int)for _, info := range prerequisites {edges[info[1]] = append(edges[info[1]], info[0]) // 邊中存的是每門課程構成的一個圖indeg[info[0]]++ // 先計算出每門課程的初始入度值,即在內層數組的下標0位置上出現一次,就代表依賴1位置的課要先上,入度就要+1}queue := []int{}for i := 0; i < numCourses; i++ {if indeg[i] == 0 {queue = append(queue, i) // 所有入度為0的入隊列}}for len(queue) > 0 {u := queue[0]queue = queue[1:]result = append(result, u) // 表示上了一門入度為0的課,看最后課的總數是否相等for _, v := range edges[u] { // 對于入度為0的數據指向的數據進行遍歷indeg[v]-- // 入度為0的數據消費了,則對應的依賴的這些節點的入度就--if indeg[v] == 0 {queue = append(queue, v)}}}return len(result) == numCourses
}