LeetCode 1514. 概率最大的路径（最短路）

题意：

给你一个由 n 个节点（下标从 0 开始）组成的无向加权图，
该图由一个描述边的列表组成，其中 edges[i] = [a, b] 表示连接节点 a 和 b 的一条无向边，
且该边遍历成功的概率为 succProb[i] 。

指定两个节点分别作为起点 start 和终点 end ，请你找出从起点到终点成功概率最大的路径，并返回其成功概率。

如果不存在从 start 到 end 的路径，请 返回 0 。只要答案与标准答案的误差不超过 1e-5 ，就会被视作正确答案。

数据范围:
2 <= n <= 10^4
0 <= start, end < n
start != end
0 <= a, b < n
a != b
0 <= succProb.length == edges.length <= 2*10^4
0 <= succProb[i] <= 1
每两个节点之间最多有一条边

解法：

令d[i]表示起点到达点i的最大路径概率.
利用最短路算法进行转移即可.

code：

const int maxm=1e4+5;
#define PI pair<int,double>
class Solution {
public:
    vector<PI>g[maxm];
    int mark[maxm];
    double d[maxm];
    double maxProbability(int n, vector<vector<int>>& e, vector<double>& cost, int st, int ed) {
        for(int i=0;i<n;i++){
            g[i].clear();
        }
        int m=e.size();
        for(int i=0;i<m;i++){
            int a=e[i][0],b=e[i][1];
            double c=cost[i];
            g[a].push_back({b,c});
            g[b].push_back({a,c});
        }
        //spfa
        queue<int>q;
        for(int i=0;i<n;i++){
            d[i]=0;
            mark[i]=0;
        }
        d[st]=1;
        q.push(st);
        mark[st]=1;
        while(q.size()){
            int x=q.front();q.pop();
            mark[x]=0;
            for(auto i:g[x]){
                int v=i.first;
                double w=i.second;
                if(d[v]<d[x]*w){
                    d[v]=d[x]*w;
                    if(!mark[v]){
                        q.push(v);
                        mark[v]=1;
                    }
                }
            }
        }
        return d[ed];
    }
};

---