Leetcode 1514. Path with Maximum Probability (python)

努利！奋斗！

已于 2022-05-21 07:20:37 修改

阅读量449

点赞数

分类专栏： Leetcode 图文章标签： python leetcode

于 2020-11-06 10:51:40 首次发布

本文链接：https://blog.csdn.net/qq_37821701/article/details/109527224

版权

这篇博客讨论了LeetCode 1514题目的解决方案，包括Bellman-Ford、Dijkstra以及一种优化后的算法。作者强调在处理概率相乘时，采用对数运算可以提高精度。在二刷题目时，通过减少空间使用，如仅用一维向量存储距离，避免使用adjacency矩阵，并利用heap实现Dijkstra算法，实现了空间复杂度的优化。

摘要由CSDN通过智能技术生成

题目

在这里插入图片描述

解法：bellman ford

这个题目本质上还是个图中求最短路径的问题
参考leetcode Path With Minimum Effort解法

class Solution:
    def maxProbability(self, n: int, edges: List[List[int]], succProb: List[float], start: int, end: int) -> float:
        graph = collections.defaultdict(list)
        for edge in edges:
            graph[edge[0]].append(edge[1])
            graph[edge[1]].append(edge[0])
        
        edge2prob = {
   }
        for i,edge in enumerate(edges):
            edge2prob[(edge[0],edge[1])] = succProb[i]
            edge2prob[(edge[1],edge[0])] = succProb[i]
        q = collections.deque()
        q.append((start,1))
        prob_vec = [0]*n
        #ans = 0
        while q:
            node,prob = q.popleft()
            for nei in graph[node]:
                new_prob = edge2prob[(nei,node)]*prob
                if new_prob>prob_vec[nei]:
                    q.append((nei,new_prob))
                    prob_vec[nei] = new_prob
        return prob_vec[end]

解法2：Dijkstra

class Solution:
    def maxProbability(self, n: int, edges: List[List[int]], succProb: List[float], start: int, end: int) -> float:
        graph = collections.defaultdict(list)
        for edge in edges:
            graph[edge[0]].append(edge[1])
            graph[edge[1]].append(edge[0])
        
        edge2prob = {
   }
        for i,edge in enumerate