题意分析
1976. 到达目的地的方案数 - 力扣(LeetCode)
这题求起始点的最短路径的个数,其实就是一个最短路+动态规划的问题。最短路径不用多说直接dijkstra,而最短路径的个数则是一个叠加过程。
可见图示:
可以发现0到3的路径总数是2,其中0-1是1条,1-2、1-3要沿袭0-1留下来的路径,各为1条,则最终到3就有2条路径,典型动态规划。
算法思路
- 实现dijkstra先求完最短路径
- 利用关键路径特点逆向记忆化搜索,不断累加各节点的路径数目。
代码实现
using ll=long long;
typedef pair<ll,int> PII;
const ll inf=LLONG_MAX/2;
const ll modi=1e9+7;
class Solution {
vector<PII> g[201];
ll dist[201];
ll pathnum[201];
public:
int countPaths(int n, vector<vector<int>>& roads){
for(int i=0;i<201;i++){
dist[i]=inf;
pathnum[i]=-1;
}
for(auto & r:roads){
int a=r[0],b=r[1],c=r[2];
g[a].emplace_back(c,b);
g[b].emplace_back(c,a);
}
priority_queue<PII,vector<PII>,greater<>> q;
q.emplace(0,0);
while(!q.empty()){
auto [d,i]=q.top();q.pop();
if(dist[i]<inf) continue;
dist[i]=d;
for(auto & [w,nxt]:g[i]){
if(dist[nxt]<inf) continue;
q.emplace(d+w,nxt);
}
}
return dfs(n-1,0);
}
ll dfs(int cur,int end){ //逆向寻找最短路径,记忆化搜索
if(cur==end) return 1;
if(pathnum[cur]!=-1) return pathnum[cur];
ll sum=0;
for(auto & [w,nxt]:g[cur]){
if(dist[cur]==w+dist[nxt]){
sum+=dfs(nxt,end);
sum%=modi;
}
}
pathnum[cur]=sum;
return sum;
}
};
解题总结
注意dijkstra和其他算法的结合,将各个板块关系套路弄明白