LeetCode题解-210/207：课程表 II/课程表

最新推荐文章于 2024-09-20 15:28:15 发布

呱圣

最新推荐文章于 2024-09-20 15:28:15 发布

阅读量159

点赞数

分类专栏： LeetCode题解文章标签： leetcode python

本文链接：https://blog.csdn.net/a250403412/article/details/107876332

版权

LeetCode题解专栏收录该内容

4 篇文章 0 订阅

订阅专栏

菜鸟成长日记

看了官方的题解，决定用广度优先搜索自己写一下
207题和210题算法一样，就是返回值不同

class Solution:
    def findOrder(self, numCourses: int, prerequisites: List[List[int]]) -> List[int]:
        def in_list(num=0, array=None): 
        	# 本想用dic的，结果获取items时提示是元组不能修改。。。还是用list吧，反正也差不多，正好课程编号是从0到n-1
        	# 将原list转换成需要的list形式[[0课程的先修课程1，0课程的先修课程2...],...]
            l = [[] for _ in range(num)]
            for item in array:
                l[item[0]].append(item[1])
            return l


        def end_study(l=None):
            # 是否全部学习完成
            for i in range(len(l)):
                if l[i] is not None:  # 有一个不是None，则还有继续学习的可能
                    return False
            return True


        def continue_study(l=None):
            # 是否能继续学习（存在[]）
            if [] in l:
                return True
            else:
                return False


        inNum_list = in_list(numCourses, prerequisites)

        order = []
        while True:  # 每次循坏表示一轮学习
            studyed = []
            if end_study(inNum_list):
                return order # 210题的返回值
                # return True # 207题的返回值
            if not continue_study(inNum_list):
                return [] # 210题的返回值
                # return False # 207题的返回值
            for i in range(len(inNum_list)):
                if inNum_list[i]==[]:
                    order.append(i)
                    studyed.append(i)
                    inNum_list[i] = None
            if studyed!=[]:
                for i in range(len(inNum_list)):
                    if inNum_list[i]:
                        for j in range(len(studyed)):
                            if studyed[j] in inNum_list[i]:
                                inNum_list[i].remove(studyed[j])