leetcode1976

题意分析

1976. 到达目的地的方案数 - 力扣（LeetCode）

        这题求起始点的最短路径的个数，其实就是一个最短路+动态规划的问题。最短路径不用多说直接dijkstra,而最短路径的个数则是一个叠加过程。

可见图示：

         可以发现0到3的路径总数是2,其中0-1是1条，1-2、1-3要沿袭0-1留下来的路径，各为1条，则最终到3就有2条路径，典型动态规划。

算法思路

1. 实现dijkstra先求完最短路径
2. 利用关键路径特点逆向记忆化搜索，不断累加各节点的路径数目。

代码实现

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和其他算法的结合，将各个板块关系套路弄明白