第2关:野外物资快速运送:王者之战,谁与争锋!

import heapq

def FindPath(nodeCount, costs, startIndex, endIndex):
    # 创建距离数组,初始时起始节点到自身的距离为0,其他节点的距离设为无穷大
    distances = [float('inf')] * nodeCount
    distances[startIndex] = 0
    
    # 创建已访问集合,记录已经找到最短路径的节点
    visited = set()
    
    # 创建优先队列,用于选择距离起始节点最近且尚未被访问的节点进行访问
    pq = [(0, startIndex)]  # 优先队列中的元素为 (距离, 节点序号)
    
    # 创建路径数组,用于记录最短路径
    path = [-1] * nodeCount
    
    while pq:
        # 从优先队列中取出距离起始节点最近的节点进行访问
        dist, node = heapq.heappop(pq)
        
        # 如果该节点已经被访问过,则跳过
        if node in visited:
            continue
        
        # 将该节点标记为已访问
        visited.add(node)
        
        # 遍历该节点的邻居节点
        for neighbor, weight in enumerate(costs[node]):
            # 如果该邻居节点未被访问,并且通过当前节点到达该邻居节点的距离更短,则更新距离数组和路径数组
            if neighbor not in visited and dist + weight < distances[neighbor]:
                distances[neighbor] = dist + weight
                heapq.heappush(pq, (dist + weight, neighbor))
                path[neighbor] = node
    
    # 根据路径数组构建最短路径
    wayList = []
    while endIndex != -1:
        wayList.append(endIndex)
        endIndex = path[endIndex]
    
    # 将最短路径逆序后返回
    return wayList[::-1]

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

YLTommi

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值