力扣 5836. 到达目的地的方案数 dijkstra+dp

129 篇文章 2 订阅
26 篇文章 0 订阅

https://leetcode-cn.com/problems/number-of-ways-to-arrive-at-destination/
在这里插入图片描述
思路:可恶啊,没开 l o n g l o n g   W A longlong\ WA longlong WA了几发。题目是求从起点到终点的最短路径的方案数,考虑在求最短路的过程中 d p dp dp,用 d p i dp_i dpi表示从起点到点 i i i的最短路径的方案数,由于此题中起点固定为0,因此可以令 d p 0 = 1 dp_0=1 dp0=1。在使用 D i j k s t r a Dijkstra Dijkstra算法求最短路的过程中,对于某一对点 ( u , v ) (u,v) (u,v)有以下几种状态:

  1. d i s [ v ] < d i s [ u ] + c o s t ( u , v ) dis[v]<dis[u]+cost(u,v) dis[v]<dis[u]+cost(u,v),即到 v v v的最短路不需要经过 u 、 v u、v uv这条边。
  2. d i s [ v ] > d i s [ u ] + c o s t ( u , v ) dis[v]>dis[u]+cost(u,v) dis[v]>dis[u]+cost(u,v),即到 v v v的最短路需要经过 u 、 v u、v uv这条边。
  3. d i s [ v ] = d i s [ u ] + c o s t ( u , v ) dis[v]=dis[u]+cost(u,v) dis[v]=dis[u]+cost(u,v),即到 v v v的最短路可以经过 u 、 v u、v uv这条边。

对于第一种状态,我们不需要做任何操作;对于第二种状态,我们计算出了一条到 v v v的新的最短路径,那么之前的 d p v dp_v dpv必然已经无效了(不满足最短路要求),因此需要令 d p v = d p u dp_v=dp_u dpv=dpu;对于第三种状态,我们发现到 v v v的最短路径有多条,之前的 d p v dp_v dpv依然有效,那么可以令 d p v = d p v + d p u dp_v=dp_v+dp_u dpv=dpv+dpu

更进一步思考,为什么这样 d p dp dp是正确的?如何保证在计算点 u u u相邻点的最短路前已经完全更新了 d p dp dp数组的?这是由最短路算法保证的。当我们第一次访问到 u u u点时,会把 ( d i s u , u ) (dis_u,u) (disu,u)放于堆中,当下一次取到 u u u时堆中所有 d i s < d i s u dis<dis_u dis<disu的状态都被更新完了,也就是说从起点到 u u u点的所有最短路径的情况都计算完了。那么此时用 d p u dp_u dpu去更新其他点当然没有任何问题。

再进一步思考,其实上一段最后几句话是有漏洞的。如果有边权为 0 0 0呢?这种情况下是不是就不能保证 d p u dp_u dpu的状态一定更新完毕了?是的,的确如此,但是不可能存在这种情况,因为如果有边权为0的话,答案应该是正无穷大。

2021.8.27更新,woc,出分发现这道题没过,仔细一看continue写成break都能system pass,真有你的啊力扣。

class Solution {
public:
    int countPaths(int n, vector<vector<int>>& roads) {
        using ll=long long;
        using pr=pair<ll,int>;
        vector<vector<int>> graph(n,vector<int>(n));
        vector<ll> dis(n,1e16),cnt(n);
        const int mod=1e9+7;
        for(vector<int>& road:roads)
            graph[road[0]][road[1]]=graph[road[1]][road[0]]=road[2];
        priority_queue<pr,vector<pr>,greater<pr>> q;
        dis[0]=0;
        cnt[0]=1;
        q.push(pr(0,0));
        while(!q.empty())
        {
            pr f=q.top();
            q.pop();
            //if(dis[f.second]<f.first)
              //  break;
              // 出大问题
            if(dis[f.second]<f.first)
                continue;
            if(f.second==n-1)
                break;
            for(int i=0;i<n;i++)
            {
                if(graph[f.second][i]&&dis[i]>=f.first+graph[f.second][i])
                {
                    if(dis[i]==f.first+graph[f.second][i])
                        cnt[i]=(cnt[i]+cnt[f.second])%mod;
                    else
                    {
                        cnt[i]=cnt[f.second];
                        q.emplace(f.first+graph[f.second][i],i);
                    }
                    dis[i]=f.first+graph[f.second][i];
                }
            }
        }
        return cnt[n-1];
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值