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)有以下几种状态:
- 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 u、v这条边。
- 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 u、v这条边。
- 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 u、v这条边。
对于第一种状态,我们不需要做任何操作;对于第二种状态,我们计算出了一条到 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];
}
};