EOJ 1818

#include <iostream>#include <algorithm>#include <cstdlib>#include <cstdio>#include <queue>#include <cstring>#include <utility>#include <vector>#include <sstream>#include <set>#define maxn 105#define INF 0x3f3f3f3fusing namespace std;typedef long long LL;int m[maxn][maxn];int d[maxn];int vis[maxn];int cnt[maxn];int num[maxn][maxn];int n;int k;void Dijkstra(){    memset(vis, 0, sizeof(vis));//vis=0标示未确定最短长度，=1标示已经确定    memset(d, INF, sizeof(d));//将最短路径数组初始化为无穷大    memset(cnt, 0, sizeof(cnt));//初始化最短路径数目数组为0    d[1] = 0;//自己对自己的距离为0    cnt[1]=1;//自己对自己的最短距离个数为1    for (int i=0; i<n; i++)    {        int small=INF;//最小值的初值为无穷大        int u = 0;//最小值下标为u        for (int j=1; j<=n; j++)        {            if(!vis[j] && d[j] <= small)                small = d[u = j];//找到比当前最小值更小的那个d值，保存坐标        }        vis[u] = 1;//此时vis[u]表示坐标为u的那个定点的最小值已经找到，放入vis数组中        for (int v=1; v<=n ; v++)        {            if (m[u][v] && !vis[v])//如果u，v中有边并且v的最小值未被确定            {                if (d[v] > d[u] + m[u][v])//如果经过u,dis[v]会更小，则交换                {                    d[v] = d[u] + m[u][v];                    cnt[v] = num[u][v] * cnt[u];//如果发现新的路径就更新cnt                }else if (d[v] == d[u] + m[u][v])                {                    cnt[v] += cnt[u] * num[u][v];//如果发现一条和原来长度相同的路径，就个数相加                }            }        }    }}int main(){    scanf("%d%d", &n, &k);    int u;    int v;    int w;    memset(m, 0, sizeof(m));//初始化保存路径长度的二维数组    for (int i=0; i<k; i++)    {        scanf("%d%d%d", &u, &v, &w);        m[u][v] = w;        num[u][v]++;//u和v中有路径，路径个数加1    }    Dijkstra();//调用函数    if (d[n] == INF)//如果不存在最短路径    {        printf("-1 0\n");    }else    {        printf("%d %d\n", d[n], cnt[n]);    }    return 0;

}

本题基本为调用上一题的函数，但是要计算最短路径的个数。每次寻找到比d还要短的路径时候，就要更替cnt的值，如果找到与d相等的路径，则要加上本次找到的路径个数。