1. 程式人生 > >圖形結構:安排課程,圖的遍歷策略

圖形結構:安排課程,圖的遍歷策略

安排課程,N門課程相互約束,prerequests = [[1],[2],[0]],上第0門課需要先上第1門課,上第1門課需要線上第2門課,上第2門課需要線上第0門課。

在這裡我們需要體會圖的遍歷,有時並不需要完全遍歷圖,遍歷圖是按照某種策略的,比如BFS策略是遍歷離當前結點最近的未遍歷的所有的點,DFS遍歷當前結點最近的一個點,在圖的遍歷中,特別是對實際問題,明確什麼是點,什麼是邊,以及遍歷的貪心策略。

程式碼實現如下:

class Soulution:
    def scheduleCourse(self,prerequests):
        num = len(
prerequests) indegree = [[],[],[],[]] # prerequests為出度表,根據出度表構建入度表,刪除某結點時,就可以刪除, # 就知道哪些結點需要刪除出度結點 for from_id in range(num): for to_id in prerequests[from_id]: indegree[to_id].append(from_id) print(indegree)
# 根據出度表,構建遍歷的佇列,遍歷的策略是,那些出度為0的點 queue =[] degree =[-1]*num for i in range(num): degree[i]= len(prerequests[i]) if degree[i] == 0: queue.append(i) # 每次彈出一個出度為0的點,然後刪除和這個點相鄰的點的出度,這個地方只需要 # 將出度的數量-1就行了 # count表示已上了幾門課
count =0 while queue: curid = queue.pop() count +=1 # 下面是更新出度,更新待上課列表的操作 for from_id in indegree[curid]: degree[from_id] -=1 if degree[from_id] == 0: queue.append(from_id) # 當上了的課程和總課程一致時代表上完了,否則代表有環,剩餘課程相互約束無法先後上 return count == num

測試結果:

prerequests = [[1],[2],[0]]
print(prerequests)
print(Soulution().scheduleCourse(prerequests))

prerequests1 = [[1],[3],[3],[]]
print(prerequests1)
print(Soulution().scheduleCourse(prerequests1))


runfile('D:/share/test/schedulleCourse.py', wdir='D:/share/test')
[[1], [2], [0]]
[[2], [0], [1], []]
False
[[1], [3], [3], []]
[[], [0], [], [1, 2]]
True