圖形結構:安排課程,圖的遍歷策略
阿新 • • 發佈:2018-12-31
安排課程,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