题解 | #【模板】单源最短路Ⅰ ‖ 无权图#(Python3)

【模板】单源最短路Ⅰ ‖ 无权图

https://www.nowcoder.com/practice/359e14832ce1476fadc70dd4bc36b991

import heapq
import sys

input = sys.stdin.read  # 读取所有输入,适用于在线评测系统

INF = int(1e18)  # 定义一个无穷大的值,用于初始化距离数组
N = int(2e6 + 5)  # 定义数组的最大长度
dist = [INF] * N  # 初始化距离数组,所有节点的距离初始为无穷大
a = [[] for _ in range(N)]  # 初始化邻接表,用于存储图


def dijkstra(s):
    """使用Dijkstra算法计算单源最短路径
    参数:
    s -- 起始节点的索引
    """
    dist[s] = 0  # 起始节点的距离设置为0
    pq = [(0, s)]  # 创建一个优先队列,用于存储待处理的节点
    heapq.heapify(pq)  # 将列表转换为堆
    while pq:
        d, u = heapq.heappop(pq)  # 弹出堆顶元素,即当前距离最小的节点
        if d > dist[u]:  # 如果当前节点的距离已经被更新过,则跳过
            continue
        for v, w in a[u]:  # 遍历当前节点的所有邻接节点
            if dist[u] + w < dist[v]:  # 如果找到更短的路径
                dist[v] = dist[u] + w  # 更新距离
                heapq.heappush(pq, (dist[v], v))  # 将新节点加入堆中


def solve():
    """解决函数,读取输入数据,调用Dijkstra算法,并输出结果"""
    data = input().split()  # 读取所有输入并分割成列表
    index = 0
    n = int(data[index])  # 读取节点数量
    index += 1
    m = int(data[index])  # 读取边的数量
    index += 1
    s = int(data[index])  # 读取起始节点
    index += 1
    for _ in range(m):
        u = int(data[index])  # 读取边的起点
        index += 1
        v = int(data[index])  # 读取边的终点
        index += 1
        a[u].append((v, 1))  # 在邻接表中添加边
    dijkstra(s)  # 调用Dijkstra算法
    result = []
    for i in range(1, n + 1):  # 遍历所有节点,准备输出结果
        if dist[i] == INF:  # 如果节点不可达,输出-1
            result.append("-1")
        else:
            result.append(str(dist[i]))  # 否则输出距离
    print(" ".join(result))  # 打印结果


def main():
    """主函数,程序入口"""
    solve()  # 调用解决函数


if __name__ == "__main__":
    main()  # 如果是直接运行脚本,则调用主函数

#15天刷题#
全部评论

相关推荐

面试官全程关摄像头1.自我介绍一下2.React和Vue哪个更熟悉一点3.你在之前那段实习经历中有没有什么技术性的突破(我只是实习了44天工作28天,我把我能说的都说了)4.你封装的哪个表单组件支不支持动态传值5.自己在实习阶段Vue3项目封装过hook吗6.hook有什么作用7.Vue2和Vue3的响应式区别(我说一个是proxy是拦截所有的底层操作,Object.defineProperty本身就是一个底层操作,有些东西拦截不了,比如数组的一些操作还有等等,面试官就说实在要拦截能不能拦截????我心想肯定不行呀,他的底层机制就不允许吧)8.pinia和vuex的区别(这个回答不出来是我太久没用了)9.pinia和zustand的区别,怎么选(直接给我干懵了)(我说react能用pinia吗&nbsp;&nbsp;他说要用的话也可以)10.渲染一万条数据,怎么解决页面卡顿问题(我说分页、监听滚轮动态加载,纯数据展示好像还可以用canvas画)(估计是没说虚拟表单,感觉不满意)11.type和interface的区别12.ts的泛型有哪些作用(我就说了一个结构相同但是类型不同的时候可以用,比如请求响应的接口,每次的data不同,这里能用一个泛型,他问我还有什么)13.你项目用的是React,如果让你再写一遍你会选择什么14.pnpm、npm、yarn的区别15.dependencies和devdependencies的区别总而言之太久没面试了,上一段实习的面试js问了很多。结果这次js一点没问,网络方面也没考,表现得很一般,但是知道自己的问题了&nbsp;&nbsp;好好准备,等待明天的影石360和周四的腾讯了&nbsp;&nbsp;加油!!!
解zj:大三的第一段面试居然是这样的结局
查看15道真题和解析
点赞 评论 收藏
分享
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务