【BFS】2023B-最长广播响应
题目描述与示例
题目描述
某通信网络中有N
个网络节点,用1
到N
进行标识。
网络中的节点互联互通,且节点之间的消息传递有时延,相连节点的时延均为一个时间单位。
现给定网络节点的连接关系link[i] = {u,v}
,其中u
和v
表示网络节点。
当指定一个节点向其他节点进行广播,所有被广播节点收到消息后都会在原路径上回复一条响应消息,请计算发送节点至少需要等待几个时间单位才能收到所有被广播节点的响应消息。
注:
-
N
的取值范围为[1,100]
;
- 连接关系
link
的长度不超过3000
,且1 <= u,v <= N
;
- 网络中任意节点间均是可达的;
输入
5 7
1 4
2 1
2 3
2 4
3 4
3 5
4 5
2
输出
4
说明
节点2
到5
的最小时延为2
,到剩余节点的最小时延均为1
,所以至少要等待2*2=4
s.
解题思路
本题也属于计算BFS层数的问题,而且只存在一个起始搜索位置,所以使用常规的单源BFS方法即可完成。
对于所给的T
行节点之间关联关系,我们需要将其转化为我们所熟悉的邻接表table
。由于本题所给的数据属于无向图,因此在构建邻接表的时候需要记录两次,即记录x
为y
的邻接节点,y
也为x
的邻接节点。代码如下
table = defaultdict(list)
for _ in range(T):
x, y = map(int, input().split())
table[x].append(y)
table[y].append(x)
剩下内容就是常规的BFS了,注意要记录层数。最终得到搜索层数为level
,答案应为(level-1)*2
。
代码
# 题目:2023B-最长广播响应
# 分值:200
# 作者:许老师-闭着眼睛学数理化
# 算法:BFS
# 代码看不懂的地方,请直接在群上提问
from collections import deque, defaultdict
# 输入节点数N,关联边数T
N, T = map(int, input().split())
# 构建无向图邻接表
table = defaultdict(list)
# 循环T次,每次输入一条边的关系
for _ in range(T):
x, y = map(int, input().split())
# x和y互为邻接边,在无向图中需要分别构建
table[x].append(y)
table[y].append(x)
# 输入起始节点
start = int(input())
# 构建检查数组checkList
# 由于数据所给的节点是1到N,故要构建长度为(N+1)的checkList
checkList = [0] * (N+1)
# 标记搜索起始节点start已经检查过
checkList[start] = 1
# 构建用于维护BFS过程的队列q
q = deque()
q.append(start)
# 构建搜索层数level
level = 0
while len(q) > 0:
# 搜索层数+1
level += 1
# 获得当前队列长度qSize
qSize = len(q)
# 当前BFS层搜索需要弹出qSize个节点
for _ in range(qSize):
# 弹出队头节点cur
cur = q.popleft()
# 考虑cur的邻接节点nxt
for nxt in table[cur]:
# 若nxt尚未被检查过
if checkList[nxt] == 0:
# 标记为已检查
checkList[nxt] = 1
# 同时加入队列q中
q.append(nxt)
# 最终的BFS搜索层数为level
# level-1为最长的边数
# (level-1)*2为最长边响应后再返回的时间,即为答案
print((level-1)*2)
时空复杂度
时间复杂度:O(N)
。每一个节点仅需遍历一次。
空间复杂度:O(N^2)
。邻接表最大占用空间。
华为OD算法/大厂面试高频题算法练习冲刺训练
-
华为OD算法/大厂面试高频题算法冲刺训练目前开始常态化报名!目前已服务100+同学成功上岸!
-
课程讲师为全网50w+粉丝编程博主@吴师兄学算法 以及小红书头部编程博主@闭着眼睛学数理化
-
每期人数维持在20人内,保证能够最大限度地满足到每一个同学的需求,达到和1v1同样的学习效果!
-
60+天陪伴式学习,40+直播课时,300+动画图解视频,300+LeetCode经典题,200+华为OD真题/大厂真题,还有简历修改、模拟面试、专属HR对接将为你解锁
-
可上全网独家的欧弟OJ系统练习华子OD、大厂真题
-
可查看链接 OD算法冲刺训练课程表 & OD真题汇总(持续更新)
-
绿色聊天软件戳 od1336
了解更多