LeetCode 1976. 到达目的地的方案数--最短路计数

1976. 到达目的地的方案数

你在一个城市里，城市由 n 个路口组成，路口编号为 0 到 n - 1 ，某些路口之间有 双向 道路。输入保证你可以从任意路口出发到达其他任意路口，且任意两个路口之间最多有一条路。

给你一个整数 n 和二维整数数组 roads ，其中 roads[i] = [ui, vi, timei] 表示在路口 ui 和 vi 之间有一条需要花费 timei 时间才能通过的道路。你想知道花费 最少时间 从路口 0 出发到达路口 n - 1 的方案数。

请返回花费 最少时间 到达目的地的 路径数目 。由于答案可能很大，将结果对 109 + 7 取余 后返回。

示例 1：

输入：n = 7, roads = [[0,6,7],[0,1,2],[1,2,3],[1,3,3],[6,3,3],[3,5,1],[6,5,1],[2,5,1],[0,4,5],[4,6,2]]
输出：4
解释：从路口 0 出发到路口 6 花费的最少时间是 7 分钟。
四条花费 7 分钟的路径分别为：

• 0 ➝ 6
• 0 ➝ 4 ➝ 6
• 0 ➝ 1 ➝ 2 ➝ 5 ➝ 6
• 0 ➝ 1 ➝ 3 ➝ 5 ➝ 6

示例 2：

输入：n = 2, roads = [[1,0,10]]
输出：1
解释：只有一条从路口 0 到路口 1 的路，花费 10 分钟。

提示：

1 <= n <= 200
n - 1 <= roads.length <= n * (n - 1) / 2
roads[i].length == 3
0 <= ui, vi <= n - 1
1 <= timei <= 109
ui != vi
任意两个路口之间至多有一条路。
从任意路口出发，你能够到达其他任意路口。

题解

直接套个迪杰斯特拉最短路模板就行，不能用佛洛依德，佛洛依德算法没法把所有情况记录下来。

AC代码

class Solution {
public:
    typedef long long ll;
    ll mod = 1000000007;
    struct Node{
        int v,next;
        ll w;//路径权重
    }edge[40010];
    int head[210],cnt;
    void add_edge(int u,int v,ll w)
    {
        edge[cnt].v=v;
        edge[cnt].w=w;
        edge[cnt].next=head[u];
        head[u]=cnt++;
    }
    void init()
    {
        memset(head,-1,sizeof(head));
        cnt=0;
    }
    ll dis[210];
    bool vis[210];
    ll num[210];
    //建立最小堆优化
    struct code{
        int u;
        ll dis;
        code(int u1,ll dis1)
        {
            u=u1;
            dis=dis1;
        }
        bool operator<(const code &t)const
        {
            return dis>t.dis;
        }
    };
    priority_queue<code>q;
    void Dij(int n)
    {
        for(int i=0;i<n;i++)
        dis[i]=2e18,vis[i]=false;
        memset(num,0,sizeof(num));
        while(!q.empty())
        q.pop();
        dis[0]=0;
        num[0]=1;
        q.push(code(0,0));
        while(!q.empty())
        {
            code t=q.top();
            q.pop();
            if(vis[t.u])continue;
            vis[t.u]=true;
            for(int i=head[t.u];i!=-1;i=edge[i].next)
            {
                int v=edge[i].v;
                int w=edge[i].w;
                //如果路径距离相等，就加上num[u]
                if(dis[v]==dis[t.u]+w)
                {
                    num[v]+=num[t.u];
                    num[v]%=mod;
                }
                //更新最短路
                else if(dis[v]>dis[t.u]+w)
                {
                    num[v]=num[t.u]%mod;
                    dis[v]=dis[t.u]+w;
                    q.push(code(v,dis[v]));
                }
            }
        }
    }
    int countPaths(int n, vector<vector<int>>& roads) 
    {
        init();
        for(int i=0;i<roads.size();i++)
        {
            add_edge(roads[i][0],roads[i][1],roads[i][2]);
            add_edge(roads[i][1],roads[i][0],roads[i][2]);
        }
        Dij(n);
        return num[n-1];
    }
};