蓝桥杯-传染病控制（python，深度优先）

最新推荐文章于 2024-10-10 15:36:50 发布

大A和小Q

最新推荐文章于 2024-10-10 15:36:50 发布

阅读量865

点赞数 1

分类专栏：蓝桥杯文章标签：蓝桥杯 python 深度优先

本文链接：https://blog.csdn.net/qq_47780707/article/details/122612528

版权

蓝桥杯专栏收录该内容

31 篇文章 2 订阅

订阅专栏

蓝桥杯-传染病控制（python）

一、题目描述

近来，一种新的传染病肆虐全球。蓬莱国也发现了零星感染者，为防止该病在蓬莱国大范围流行，该国政府决定不惜一切代价控制传染病的蔓延。不幸的是，由于人们尚未完全认识这种传染病，难以准确判别病毒携带者，更没有研制出疫苗以保护易感人群。于是，蓬莱国的疾病控制中心决定采取切断传播途径的方法控制疾病传播。经过 WHO（世界卫生组织）以及全球各国科研部门的努力，这种新兴传染病的传播途径和控制方法已经研究消楚，剩下的任务就是由你协助蓬莱国疾控中心制定一个有效的控制办法。
问题描述
研究表明，这种传染病的传播具有两种很特殊的性质；
第一是它的传播途径是树型的，一个人X只可能被某个特定的人Y感染，只要Y不得病，或者是XY之间的传播途径被切断，则X就不会得病。
第二是，这种疾病的传播有周期性，在一个疾病传播周期之内，传染病将只会感染一代患者，而不会再传播给下一代。
这些性质大大减轻了蓬莱国疾病防控的压力，并且他们已经得到了国内部分易感人群的潜在传播途径图（一棵树）。但是，麻烦还没有结束。由于蓬莱国疾控中心人手不够，同时也缺乏强大的技术，以致他们在一个疾病传播周期内，只能设法切断一条传播途径，而没有被控制的传播途径就会引起更多的易感人群被感染（也就是与当前已经被感染的人有传播途径相连，且连接途径没有被切断的人群）。当不可能有健康人被感染时，疾病就中止传播。所以，蓬莱国疾控中心要制定出一个切断传播途径的顺序，以使尽量少的人被感染。你的程序要针对给定的树，找出合适的切断顺序。
输入
输入格式的第一行是两个整数n（1≤n≤300）和p。接下来p行，每一行有两个整数i和j，表示节点i和j间有边相连（意即，第i人和第j人之间有传播途径相连，注意：可能是i到j也可能是j到i）。其中节点1是已经被感染的患者。
对于给定的输入数据，如果不切断任何传播途径，则所有人都会感染。
输出
只有一行，输出总共被感染的人数。
样例输入

样例输出

二、解题思路与代码

将输入转化为树形图，用一个二维数组表示，其中i结点的子节点为root[i]数组。然后采取深度优先的思路模拟剪去每一层的某个传播途径，用当前已传染人数和当前最优传染人数剪枝。用递归函数实现深度优先。

# 用深度优先算法，用链接表的方式存储,再转化成树，剪枝的话要剪得越浅越好（差别会很大）

n, p = map(int, input().split())
# 链接表的第一个数组不用
linkedlist = [[] for _ in range(n+1)]
for _ in range(p):
    x, y = map(int, input().split())
    linkedlist[x].append(y)
    linkedlist[y].append(x)

# 从链接表构造树，因为直接构造树会导致多次遍历输入数组效率低
tree = [[] for _ in range(n+1)]
# 构造树其中i结点的子节点为root[i]
def createTree(root):
    if len(linkedlist[root])!=0:
        tree[root] = linkedlist[root][:]
        for x in linkedlist[root]:
            linkedlist[x].remove(root)
            createTree(x)
createTree(1)
        


# 输入当前层结点和当前感染人数，计算从下一层开始的被感染的最少人数并返回人数
bestresult = 300
def dfs(currentlayer, currentcount):
    global tree, bestresult
    nextlayer = []
    for x in currentlayer:
        nextlayer += tree[x]
    minnum = 300
    # 到达叶子结点，更新bestresult
    if nextlayer ==[]:
        if currentcount<bestresult:
            bestresult = currentcount
        return 0
    # 若感染人数大于bestresult则剪枝
    elif currentcount + len(nextlayer)-1<=bestresult:
        for i,x in enumerate(nextlayer):
            del nextlayer[i]
            nextnum = dfs(nextlayer, currentcount + len(nextlayer)-1)
            if minnum>nextnum:
                minnum = nextnum
            nextlayer.insert(i,x)
        return len(nextlayer)-1+minnum
    # 若大于则直接返回300（表示剪枝）
    else:
        return 300


#最后加的1是加上原本就被感染的根节点
print(dfs([1], 1)+1)