leetcode1514

刷题主页

给你一个由 n 个节点（下标从 0 开始）组成的无向加权图，该图由一个描述边的列表组成，其中 edges[i] = [a, b] 表示连接节点 a 和 b 的一条无向边，且该边遍历成功的概率为 succProb[i] 。
指定两个节点分别作为起点 start 和终点 end ，请你找出从起点到终点成功概率最大的路径，并返回其成功概率。
如果不存在从 start 到 end 的路径，请 返回 0 。只要答案与标准答案的误差不超过 1e-5 ，就会被视作正确答案。

输入：n = 3, edges = [[0,1],[1,2],[0,2]], succProb = [0.5,0.5,0.2], start = 0, end = 2
输出：0.25000
解释：从起点到终点有两条路径，其中一条的成功概率为 0.2 ，而另一条为 0.5 * 0.5 = 0.25

输入：n = 3, edges = [[0,1],[1,2],[0,2]], succProb = [0.5,0.5,0.3], start = 0, end = 2
输出：0.30000

输入：n = 3, edges = [[0,1]], succProb = [0.5], start = 0, end = 2
输出：0.00000
解释：节点 0 和 节点 2 之间不存在路径

悲催的周赛，很经典的无向图问题，然而自己做的有关图的题目太少了，这道题还是WA了，在此记录下。无向图问题经典套路是进行BFS遍历，同时需要使用邻接表记录每条边连接哪两个点，之后根据题目要求进行选择即可。

class Solution {
public:
    double maxProbability(int n, vector<vector<int>>& edges, vector<double>& succProb, int start, int end) {
    //建立邻接表，这步在无向图中通常是必须的，因为需要根据其进行BFS遍历
    //套路也就是下面所示，记录每个结点和其它哪些结点相连。
        vector<pair<int,double>> Edge[n];
        for(int i=0;i<edges.size();++i){
            Edge[edges[i][0]].push_back({edges[i][1],succProb[i]});
            Edge[edges[i][1]].push_back({edges[i][0],succProb[i]});
        }
        //队列进行BFS
        queue<int>que;
        //因为是无向图，为了避免死循环，需要记录哪些点遍历过
        vector<bool>visited(n,false);
        //记录每个节点的结果
        vector<double>prob(n,0);
        que.push(start);
        visited[start]=true;
        prob[start]=1.0;
        //经典BFS模板
        while(!que.empty()){
            int temp=que.front();
            que.pop();
            visited[temp]=false;
            double cur=prob[temp];
            //遍历和该点连接的下一个结点
            for(const auto& e:Edge[temp]){
                int to=e.first;
                double p=e.second;
                //根据题目判断选择
                if(cur*p>prob[to]){
                    prob[to]=cur*p;
                    //该点入队了，避免再次遍历到该点，让visited为真
                    if(!visited[to])que.push(to),visited[to]=true;
                }
            }
        }
        return prob[end];
    }
};