LeetCode 207. 課程表 | Python

207. 課程表


題目來源:力扣(LeetCode)https://leetcode-cn.com/problems/course-schedule

題目


你這個(gè)學(xué)期必須選修 numCourse 門課程,記為 0 到 numCourse-1 柄错。

在選修某些課程之前需要一些先修課程嚎尤。 例如,想要學(xué)習(xí)課程 0 酥泞,你需要先完成課程 1 砚殿,我們用一個(gè)匹配來表示他們:[0,1]

給定課程總量以及它們的先決條件,請你判斷是否可能完成所有課程的學(xué)習(xí)芝囤?

示例 1:

輸入: 2, [[1,0]]
輸出: true
解釋: 總共有 2 門課程似炎。學(xué)習(xí)課程 1 之前,你需要完成課程 0悯姊。所以這是可能的羡藐。

示例 2:

輸入: 2, [[1,0],[0,1]]
輸出: false
解釋: 總共有 2 門課程。學(xué)習(xí)課程 1 之前悯许,你需要先完成?課程 0仆嗦;并且學(xué)習(xí)課程 0 之前,你還應(yīng)先完成課程 1岸晦。這是不可能的欧啤。

提示:

  1. 輸入的先決條件是由 邊緣列表 表示的圖形,而不是 鄰接矩陣 启上。詳情請參見圖的表示法邢隧。
  2. 你可以假定輸入的先決條件中沒有重復(fù)的邊。
  3. 1 <= numCourses <= 10^5

解題思路


思路:拓?fù)渑判颍˙FS冈在,DFS)

其實(shí)倒慧,這是一道經(jīng)典的【拓?fù)渑判颉繂栴}。

首先先審題,結(jié)合示例 1 和示例 2纫谅,我們其實(shí)可以看到炫贤,其實(shí)題目問的是給定輸入先決條件表示的圖形(也就是課程表)是否是有向無環(huán)圖。也是說付秕,當(dāng)確定先決條件之后兰珍,圖形不能存在環(huán),否則不成立询吴。

有向無環(huán)圖(DAG):指的一個(gè)有向圖從任意頂點(diǎn)出發(fā)無法經(jīng)過若干條邊回到該點(diǎn)掠河,則該圖是一個(gè)有向無環(huán)圖。

有向無環(huán)圖猛计,詳細(xì)信息可由下方入口進(jìn)行了解
https://en.wikipedia.org/wiki/Directed_acyclic_graph (維基百科:有向無環(huán)圖)
https://baike.baidu.com/item/%E6%9C%89%E5%90%91%E6%97%A0%E7%8E%AF%E5%9B%BE/10972513?fr=aladdin (百度百科:有向無環(huán)圖)

那么當(dāng)要求課程安排圖是否是有向無環(huán)圖時(shí)唠摹,我們用拓?fù)渑判騺砼袛唷M負(fù)渑判颍簩D所有頂點(diǎn)進(jìn)行排序成線性序列奉瘤,使得圖中任意一個(gè)有向邊 uv勾拉,u 在線性序列中出現(xiàn)在 v 前面。

拓?fù)渑判虻廖拢敿?xì)信息可由下方入口進(jìn)行了解
https://en.wikipedia.org/wiki/Topological_sorting (維基百科:拓?fù)渑判颍?br> https://baike.baidu.com/item/%E6%8B%93%E6%89%91%E6%8E%92%E5%BA%8F/5223807?fr=aladdin (百度百科:拓?fù)渑判颍?/p>

在題目中的提示中有說明藕赞,輸入的先決條件時(shí)由 邊緣列表 表示,在這里肌访,我們要將此轉(zhuǎn)換為 鄰接表找默。

鄰接表,詳細(xì)信息可由下方入口進(jìn)行了解
https://en.wikipedia.org/wiki/Adjacency_list (維基百科:鄰接表)
https://baike.baidu.com/item/%E9%82%BB%E6%8E%A5%E8%A1%A8/9796152?fr=aladdin (百度百科:鄰接表)

廣度優(yōu)先搜索(BFS)

首先吼驶,先將邊緣列表表示的先決條件轉(zhuǎn)換得到鄰接表惩激。然后統(tǒng)計(jì)圖中每個(gè)節(jié)點(diǎn)的入度情況,存儲在一個(gè)列表中蟹演。(當(dāng)入度為 0 時(shí)风钻,表示點(diǎn)不作為任何邊的終點(diǎn),也就說該點(diǎn)是所有連接邊的起點(diǎn))

在這里酒请,定義輔助隊(duì)列 queue骡技,將所有入度為 0 的節(jié)點(diǎn)存入隊(duì)列中,用于后續(xù)處理羞反。

開始進(jìn)行搜索布朦,令隊(duì)列中的節(jié)點(diǎn)出隊(duì):

  • 將當(dāng)前節(jié)點(diǎn)鄰接節(jié)點(diǎn)入度減 1;
  • 當(dāng)鄰接節(jié)點(diǎn)入度為 0 時(shí)昼窗,將其入隊(duì)是趴。

當(dāng)隊(duì)列中的節(jié)點(diǎn)出隊(duì)時(shí),令課程總量減 1:

  • 如果課程圖是有向無環(huán)圖澄惊,若完成拓?fù)渑判蛩敉荆敲此械墓?jié)點(diǎn)都會入隊(duì)出隊(duì)富雅;
  • 若是課程圖存在環(huán),那么一定存在節(jié)點(diǎn)入度不為 0 的情況肛搬;
  • 也就是說没佑,當(dāng)所有節(jié)點(diǎn)入隊(duì)出隊(duì)后,課程總量為 0 的情況下温赔,也就能證明課程圖是否是有向無環(huán)圖蛤奢。

具體的代碼見【代碼實(shí)現(xiàn) # 廣度優(yōu)先搜索】

深度優(yōu)先搜索(DFS)

我們先看使用深度優(yōu)先搜索的思路來判斷圖是否有環(huán)。

在這里让腹,我們借助一個(gè)輔助列表(當(dāng)然也可以考慮用棧)標(biāo)記每個(gè)節(jié)點(diǎn)的狀態(tài):

  • 未標(biāo)記远剩,令此時(shí)狀態(tài)為 0;
  • 臨時(shí)標(biāo)記骇窍,令此時(shí)狀態(tài)為 1;
  • 永久標(biāo)記锥余,令此時(shí)狀態(tài)為 -1腹纳。

開始對每個(gè)節(jié)點(diǎn)進(jìn)行深度優(yōu)先搜索,當(dāng)存在環(huán)時(shí)驱犹,返回 False嘲恍,執(zhí)行過程如下:

  • 當(dāng)節(jié)點(diǎn)狀態(tài)為 -1 時(shí),表示已永久標(biāo)記雄驹,此時(shí)已搜索完畢佃牛,不許重復(fù)搜索,直接返回 True医舆;
  • 當(dāng)節(jié)點(diǎn)狀態(tài)為 1 時(shí)俘侠,表示臨時(shí)標(biāo)記,也就說此節(jié)點(diǎn)蔬将,在此次深搜中又一次對該節(jié)點(diǎn)進(jìn)行了搜索爷速,那么表示圖中存在環(huán),那么直接返回 False霞怀;
  • 當(dāng)狀態(tài)為 0 時(shí)惫东,先將節(jié)點(diǎn)臨時(shí)標(biāo)記(令狀態(tài)為 1),然后以此點(diǎn)繼續(xù)搜索毙石,直至遇到終止條件廉沮。
  • 當(dāng)節(jié)點(diǎn)搜索完畢,未發(fā)現(xiàn)閉環(huán)徐矩,則去除臨時(shí)標(biāo)記滞时,將節(jié)點(diǎn)進(jìn)行永久標(biāo)記(令狀態(tài)為 -1)

如果所有節(jié)點(diǎn)都搜索完畢后,不存在環(huán)丧蘸,則返回 True漂洋。

具體的代碼見【代碼實(shí)現(xiàn) # 深度優(yōu)先搜索】

代碼實(shí)現(xiàn)


# 廣度優(yōu)先搜索
class Solution:
    def canFinish(self, numCourses: int, prerequisites: List[List[int]]) -> bool:
        from collections import deque

        # 將邊緣列表表示的先決條件轉(zhuǎn)化為 鄰接表
        adjacency = [[] for _ in range(numCourses)]
        # 定義列表統(tǒng)計(jì)圖中每個(gè)節(jié)點(diǎn)的入度情況
        indegree = [0] * numCourses

        for info in prerequisites:
            adjacency[info[1]].append(info[0])
            indegree[info[0]] += 1
        
        queue = deque()

        # 將入度為 0 的節(jié)點(diǎn)入隊(duì)
        for i in range(len(indegree)):
            if not indegree[i]:
                queue.append(i)
        # 開始進(jìn)行搜索
        while queue:
            u = queue.popleft()
            numCourses -= 1
            # 搜索鄰接節(jié)點(diǎn)
            for v in adjacency[u]:
                # 將鄰接節(jié)點(diǎn)入度減 1
                indegree[v] -= 1
                # 如果入度為 0遥皂,入隊(duì)
                if indegree[v] == 0:
                    queue.append(v)

        return numCourses == 0

# 深度優(yōu)先搜索
class Solution:
    def canFinish(self, numCourses: int, prerequisites: List[List[int]]) -> bool:
        def dfs(adjacency, sign, i):
            if sign[i] == -1:
                return True
            if sign[i] == 1:
                return False
            # 開始搜索,先進(jìn)行臨時(shí)標(biāo)記
            sign[i] = 1
            # 以此節(jié)點(diǎn)往下搜索
            for j in adjacency[i]:
                if not dfs(adjacency, sign, j):
                    return False
            sign[i] = -1
            return True

        # 定義輔助列表標(biāo)記狀態(tài)刽漂,初始化為 0演训,表示未標(biāo)記
        sign = [0] * numCourses
        # 將邊緣列表表示的先決條件轉(zhuǎn)化為 鄰接表
        adjacency = [[] for _ in range(numCourses)]
        for info in  prerequisites:
            adjacency[info[1]].append(info[0])
        
        # 開始深搜
        for i in range(numCourses):
            if not dfs(adjacency, sign, i):
                return False
        return True

實(shí)現(xiàn)結(jié)果


實(shí)現(xiàn)結(jié)果 | 廣度優(yōu)先搜索
實(shí)現(xiàn)結(jié)果 | 深度優(yōu)先搜索

歡迎關(guān)注


公眾號 【書所集錄

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市贝咙,隨后出現(xiàn)的幾起案子样悟,更是在濱河造成了極大的恐慌,老刑警劉巖庭猩,帶你破解...
    沈念sama閱讀 217,509評論 6 504
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件窟她,死亡現(xiàn)場離奇詭異,居然都是意外死亡蔼水,警方通過查閱死者的電腦和手機(jī)震糖,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,806評論 3 394
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來趴腋,“玉大人吊说,你說我怎么就攤上這事∮啪妫” “怎么了颁井?”我有些...
    開封第一講書人閱讀 163,875評論 0 354
  • 文/不壞的土叔 我叫張陵,是天一觀的道長蠢护。 經(jīng)常有香客問我雅宾,道長,這世上最難降的妖魔是什么葵硕? 我笑而不...
    開封第一講書人閱讀 58,441評論 1 293
  • 正文 為了忘掉前任眉抬,我火速辦了婚禮,結(jié)果婚禮上贬芥,老公的妹妹穿的比我還像新娘吐辙。我一直安慰自己,他們只是感情好蘸劈,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,488評論 6 392
  • 文/花漫 我一把揭開白布昏苏。 她就那樣靜靜地躺著,像睡著了一般威沫。 火紅的嫁衣襯著肌膚如雪贤惯。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 51,365評論 1 302
  • 那天棒掠,我揣著相機(jī)與錄音孵构,去河邊找鬼。 笑死烟很,一個(gè)胖子當(dāng)著我的面吹牛颈墅,可吹牛的內(nèi)容都是我干的蜡镶。 我是一名探鬼主播,決...
    沈念sama閱讀 40,190評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼恤筛,長吁一口氣:“原來是場噩夢啊……” “哼官还!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起毒坛,我...
    開封第一講書人閱讀 39,062評論 0 276
  • 序言:老撾萬榮一對情侶失蹤望伦,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后煎殷,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體屯伞,經(jīng)...
    沈念sama閱讀 45,500評論 1 314
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,706評論 3 335
  • 正文 我和宋清朗相戀三年豪直,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了劣摇。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 39,834評論 1 347
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡顶伞,死狀恐怖饵撑,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情唆貌,我是刑警寧澤,帶...
    沈念sama閱讀 35,559評論 5 345
  • 正文 年R本政府宣布垢乙,位于F島的核電站锨咙,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏追逮。R本人自食惡果不足惜酪刀,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,167評論 3 328
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望钮孵。 院中可真熱鬧骂倘,春花似錦、人聲如沸巴席。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,779評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽漾唉。三九已至荧库,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間赵刑,已是汗流浹背分衫。 一陣腳步聲響...
    開封第一講書人閱讀 32,912評論 1 269
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留般此,地道東北人蚪战。 一個(gè)月前我還...
    沈念sama閱讀 47,958評論 2 370
  • 正文 我出身青樓牵现,卻偏偏與公主長得像,于是被迫代替她去往敵國和親邀桑。 傳聞我的和親對象是個(gè)殘疾皇子瞎疼,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,779評論 2 354