广度优先遍历-选课问题

最新推荐文章于 2022-05-14 09:32:50 发布

wxdoop

最新推荐文章于 2022-05-14 09:32:50 发布

阅读量144

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/qq_36235098/article/details/103156747

版权

算法专栏收录该内容

33 篇文章 0 订阅

订阅专栏

# 题目选课
# 数学-》
# 计算机基础 -》 java - 》
#               英语 - 》算法
#
# 怎样选课？


# 创建一个列表表示课程之间的关系，1表示依赖

class_relation = [
    [0,0,1,0,0],
    [0,0,1,0,0],
    [0,0,0,0,1],
    [0,0,0,0,1],
    [0,0,0,0,0]
]

def choose_class(class_num,class_relation):
    # 课程的依赖数
    class_needs = [0 for i in range(class_num)]
    for i in range(len(class_relation)):
        for j in range(len(class_relation[i])):
            if class_relation[i][j] ==1:
                class_needs[j]+=1
    # 可以直接选的课程,元素为课程号
    class_first_choose = [i for i in range(len(class_needs)) if class_needs[i]==0]
	# 结果
    res = []
    while len(class_first_choose)!=0:
        res.append(class_first_choose[0])
        class_id = class_first_choose.pop(0)
        for i in range(len(class_relation[class_id])):
            if class_relation[class_id][i] == 1:
                class_needs[i]-=1
                if class_needs[i]==0:
                    class_first_choose.append(i)
        print(res)

choose_class(5,class_relation)